Missed one CHECK_NOTNULL in pytorch#82032's find-replace

cchan · web-flow · commit ccb82d676755 · 2022-09-08T14:39:07.000-07:00
diff --git a/caffe2/contrib/nccl/cuda_nccl_gpu.cc b/caffe2/contrib/nccl/cuda_nccl_gpu.cc
@@ -177,7 +177,7 @@ void runNCCL(const NCCLExecution& ex, InitF&& init_f, F&& f) {
   // Now, wait on all the events in the original stream.
   CUDAGuard dg(ex.stream_gpu_id);
   for (auto& event : events) {
-    CUDA_ENFORCE(cudaStreamWaitEvent(CHECK_NOTNULL(ex.stream), event, 0));
+    CUDA_ENFORCE(cudaStreamWaitEvent(TORCH_CHECK_NOTNULL(ex.stream), event, 0));
   }
 }
 

Original file line number	Diff line number	Diff line change
`@@ -177,7 +177,7 @@ void runNCCL(const NCCLExecution& ex, InitF&& init_f, F&& f) {`
`177`	`177`	`// Now, wait on all the events in the original stream.`
`178`	`178`	`CUDAGuard dg(ex.stream_gpu_id);`
`179`	`179`	`for (auto& event : events) {`
`180`		`- CUDA_ENFORCE(cudaStreamWaitEvent(CHECK_NOTNULL(ex.stream), event, 0));`
	`180`	`+ CUDA_ENFORCE(cudaStreamWaitEvent(TORCH_CHECK_NOTNULL(ex.stream), event, 0));`
`181`	`181`	`}`
`182`	`182`	`}`
`183`	`183`