From fae61d3ef772aa14b41f425cf3851c7bfa7fdaf6 Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Thu, 4 Jun 2026 04:13:40 +0000
Subject: [PATCH] Add c10/cuda/CUDAStream.h include for getCurrentCUDAStream

---
 dsv4/kernels/cuda/blackwell_swizzle.cu | 1 +
 1 file changed, 1 insertion(+)

diff --git a/dsv4/kernels/cuda/blackwell_swizzle.cu b/dsv4/kernels/cuda/blackwell_swizzle.cu
index 1fb77914..5c6f1479 100644
--- 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