Fix missing #endif after rebase

Signed-off-by: Serge Panev <[email protected]>
apache · Sep 7, 2021 · d863554 · d863554
1 parent 362f56a
commit d863554
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/ndarray/ndarray.cc b/src/ndarray/ndarray.cc
@@ -1189,10 +1189,9 @@ void SetValueOp(const real_t& rhs, NDArray* out) {
             } else {
               ndarray::Eval(ctx.get_stream<gpu>(), rhs, ret);
             }
-            // Wait GPU kernel to complete
-            ctx.get_stream<gpu>()->Wait();
             break;
           }
+#endif
           default:
             LOG(FATAL) << MXNET_GPU_NOT_ENABLED_ERROR;
         }