Add c10/cuda/CUDAStream.h include for getCurrentCUDAStream

2026-06-04 04:13:40 +00:00
parent ee86969f6c
commit fae61d3ef7
1 changed files with 1 additions and 0 deletions
--- a/dsv4/kernels/cuda/blackwell_swizzle.cu
+++ b/dsv4/kernels/cuda/blackwell_swizzle.cu
@@ -11,6 +11,7 @@
 */

 #include <cuda_runtime.h>
+#include <c10/cuda/CUDAStream.h>
 #include <cstdint>
 #include <torch/extension.h>  // For pybind11 bindings