c_api.h

Go to the documentation of this file.

 
#ifndef RABIT_C_API_H_
#define RABIT_C_API_H_
 
#ifdef __cplusplus
#define RABIT_EXTERN_C extern "C"
#include <cstdio>
#else
#define RABIT_EXTERN_C
#include <stdio.h>
#endif  // __cplusplus
 
#if defined(_MSC_VER) || defined(_WIN32)
#define RABIT_DLL RABIT_EXTERN_C __declspec(dllexport)
#else
#define RABIT_DLL RABIT_EXTERN_C __attribute__ ((visibility ("default")))
#endif  // defined(_MSC_VER) || defined(_WIN32)
 
typedef unsigned long rbt_ulong;  // NOLINT(*)
 
RABIT_DLL bool RabitInit(int argc, char *argv[]);
 
RABIT_DLL int RabitFinalize(void);
 
RABIT_DLL int RabitGetRingPrevRank(void);
 
RABIT_DLL int RabitGetRank(void);
 
RABIT_DLL int RabitGetWorldSize(void);
 
RABIT_DLL int RabitIsDistributed(void);
 
RABIT_DLL int RabitTrackerPrint(const char *msg);
RABIT_DLL void RabitGetProcessorName(char *out_name,
                                     rbt_ulong *out_len,
                                     rbt_ulong max_len);
RABIT_DLL int RabitBroadcast(void *sendrecv_data, rbt_ulong size, int root);
 
RABIT_DLL int RabitAllgather(void *sendrecvbuf, size_t total_size,
                             size_t beginIndex, size_t size_node_slice,
                             size_t size_prev_slice, int enum_dtype);
 
RABIT_DLL int RabitAllreduce(void *sendrecvbuf, size_t count, int enum_dtype,
                             int enum_op, void (*prepare_fun)(void *arg),
                             void *prepare_arg);
 
RABIT_DLL int RabitVersionNumber(void);
 
 
RABIT_DLL int RabitLinkTag(void);
 
#endif  // RABIT_C_API_H_