mxnet 源碼閱讀 8

標籤：sync   handle   net   src   led   let   shu   variable   shutdown   

mxnet/include/mxnet/engine.h

在名字空間 mxnet engine 中定義了抽象類別Engine，用以規範介面，其介面有

NotifyShutdown

NewVariable

DeleteVariable

NewOperator

DeleteOperator

Push

PushAsync

PushSync

WaitForVar

WaitForAll

DeduplicateVarHandle

等

mxnet/src/engine/engine_impl.h

定義了多態實現的

Engine *CreateNaiveEngine

Engine *CreateThreadedEnginePolled

Engine *CreateThreadedEnginePerDevice

 mxnet/src/engine/engine.cc

根據環境變數決定Engine *CreateEngine的選擇情況。有三種形式

CreateNaiveEngine，CreateThreadedEnginePooled，CreateThreadEnginePerDevice。

 

mxnet 源碼閱讀 8