Commit 956d3aef authored by Rick Ho's avatar Rick Ho
Browse files

fix stream manager

parent bb3c8966
...@@ -24,7 +24,7 @@ void CudaStreamManager::sync(int idx) { ...@@ -24,7 +24,7 @@ void CudaStreamManager::sync(int idx) {
} }
void CudaStreamManager::setup(const int device) { void CudaStreamManager::setup(const int device) {
#ifdef MOE_USE_NCCL #ifdef FMOE_USE_NCCL
this->ncclgood = 0; this->ncclgood = 0;
#endif #endif
this->device = device; this->device = device;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment