ML.DMLC.XGBoost.Rabit.FFI

Documentation

c_rabitInit :: Int32 -> StringArray -> IO ()

c_rabitFinalize :: IO ()

c_rabitGetRank :: IO Int32

c_rabitGetWorldSize :: IO Int32

c_rabitIsDistributed :: IO Int32

c_rabitTrackerPrint :: StringPtr -> IO ()

c_rabitGetProcessorName :: StringPtr -> Ptr CULong -> CULong -> IO ()

c_rabitBroadcast :: Ptr a -> CULong -> Int32 -> IO ()

c_rabitAllreduce :: Ptr a -> CSize -> Int32 -> Int32 -> Ptr () -> Ptr () -> IO ()

c_rabitLoadCheckPoint :: Ptr StringPtr -> Ptr CULong -> Ptr StringPtr -> Ptr CULong -> IO Int32

c_rabitCheckPoint :: StringPtr -> CULong -> StringPtr -> CULong -> IO ()

c_rabitVersionNumber :: IO Int32

c_rabitLinkTag :: IO Int32

rabitInit :: [String] -> IO ()

rabitFinalize :: IO ()

rabitGetRank :: IO Int32

rabitGetWordSize :: IO Int32

rabitIsDistributed :: IO Bool

rabitTrackerPrint :: String -> IO ()

rabitGetProcessorName :: IO String

rabitBoradcast

Arguments

:: Ptr a

the pointer to send or recive buffer.

-> Int32

the size of data

-> Int32

the root of process

-> IO ()

data AllreduceOpType

Ref: https://github.com/dmlc/rabit/blob/master/include/rabit/internal/engine.h#L172

Constructors

KMax

KMin

KSum

KBitwiseOR

Instances

Enum AllreduceOpType

Methods

succ :: AllreduceOpType -> AllreduceOpType

pred :: AllreduceOpType -> AllreduceOpType

toEnum :: Int -> AllreduceOpType

fromEnum :: AllreduceOpType -> Int

enumFrom :: AllreduceOpType -> [AllreduceOpType]

enumFromThen :: AllreduceOpType -> AllreduceOpType -> [AllreduceOpType]

enumFromTo :: AllreduceOpType -> AllreduceOpType -> [AllreduceOpType]

enumFromThenTo :: AllreduceOpType -> AllreduceOpType -> AllreduceOpType -> [AllreduceOpType]

Eq AllreduceOpType

Methods

(==) :: AllreduceOpType -> AllreduceOpType -> Bool

(/=) :: AllreduceOpType -> AllreduceOpType -> Bool

data AllreduceDataType

Ref: https://github.com/dmlc/rabit/blob/master/include/rabit/internal/engine.h#L179

Constructors

KChar

KUChar

KInt

KUInt

KLong

KULong

KFloat

KDouble

KLongLong

KULongLong

Instances

Enum AllreduceDataType

Methods

succ :: AllreduceDataType -> AllreduceDataType

pred :: AllreduceDataType -> AllreduceDataType

toEnum :: Int -> AllreduceDataType

fromEnum :: AllreduceDataType -> Int

enumFrom :: AllreduceDataType -> [AllreduceDataType]

enumFromThen :: AllreduceDataType -> AllreduceDataType -> [AllreduceDataType]

enumFromTo :: AllreduceDataType -> AllreduceDataType -> [AllreduceDataType]

enumFromThenTo :: AllreduceDataType -> AllreduceDataType -> AllreduceDataType -> [AllreduceDataType]

Eq AllreduceDataType

Methods

(==) :: AllreduceDataType -> AllreduceDataType -> Bool

(/=) :: AllreduceDataType -> AllreduceDataType -> Bool

rabitAllreduce

Arguments

:: Ptr a

buffer for both sending and recving data

-> Int32

number of elements to be reduced

-> AllreduceDataType

-> AllreduceOpType

-> IO ()

rabitVersionNumber :: IO Int32

rabitLinkTag :: IO Int32