Add C10_CUDA_KERNEL_LAUNCH_CHECK

pytorch · Apr 28, 2024 · 33ee903 · 33ee903
1 parent b34eac4
commit 33ee903
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch/csrc/distributed/c10d/quantization/quantization_gpu.cu b/torch/csrc/distributed/c10d/quantization/quantization_gpu.cu
@@ -99,7 +99,7 @@ at::Tensor _float_to_bfloat16_cuda(const at::Tensor& input) {
 #else
       reinterpret_cast<uint16_t*>(output.mutable_data_ptr<at::Half>()));
 #endif
-  //C10_CUDA_KERNEL_LAUNCH_CHECK();
+  C10_CUDA_KERNEL_LAUNCH_CHECK();
 
   return output;
 }