fix stream manager

956d3aef · Rick Ho · bb3c8966 · 956d3aef
Commit 956d3aef authored Apr 26, 2021 by Rick Ho
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

cuda/stream_manager.cpp cuda/stream_manager.cpp +1 -1

No files found.
--- a/cuda/stream_manager.cpp
+++ b/cuda/stream_manager.cpp
@@ -24,7 +24,7 @@ void CudaStreamManager::sync(int idx) {
 }
 void CudaStreamManager::setup(const int device) {
-#ifdef MOE_USE_NCCL
+#ifdef FMOE_USE_NCCL
    this->ncclgood = 0;
 #endif
    this->device = device;