Bugfix: memsetAsync uses wrong default stream

Signed-off-by: Peter Han <fujun.han@iluvatar.ai>
2021-03-23 21:11:42 +08:00
parent 50bf00e5f2
commit 92393b2676
8 changed files with 16 additions and 16 deletions
@@ -244,7 +244,7 @@ public:
    void *workspace = nullptr, 
    cudaStream_t stream = nullptr) {
    
-    Status status = initialize(args, workspace);
+    Status status = initialize(args, workspace, stream);
    
    if (status == Status::kSuccess) {
      status = run(stream);