test_graph_cudnn_dropout now passes too

mcarilli · mcarilli · commit bdc6d300b514 · 2022-01-18T11:31:18.000-07:00
diff --git a/c10/cuda/CUDAMallocAsyncAllocator.cpp b/c10/cuda/CUDAMallocAsyncAllocator.cpp
@@ -270,7 +270,7 @@ void free(void* ptr) {
                       "This may be benign, for example, a Python tensor in the capture "
                       "might happen to shadow (use the same name as) an unrelated temporary "
                       "tensor from somewhere before capture, pushing the earlier tensor "
-                      "out of scope.\n"
+                      "out of scope. "
                       "However, if the tensor we're freeing here IS used by the capture, "
                       "freeing it is an error, and may cause illegal memory accesses or "
                       "memory corruption during graph replay.");
diff --git a/test/test_cuda.py b/test/test_cuda.py
@@ -3711,6 +3711,8 @@ def test_graph_cudnn_dropout(self):
             g.capture_end()
         torch.cuda.current_stream().wait_stream(s)
 
+        g.replay()
+
         y = model(x)
 
     @unittest.skipIf((not TEST_CUDA) or
