Update convolution_gather_scatter_cuda.cu

Deathdadev · web-flow · commit bebadc476e92 · 2025-06-22T23:06:20.000+10:00
diff --git a/torchsparse/backend/convolution/convolution_gather_scatter_cuda.cu b/torchsparse/backend/convolution/convolution_gather_scatter_cuda.cu
@@ -1,6 +1,7 @@
 #include <ATen/cuda/CUDAContext.h>
 #include <cublas_v2.h>
 #include <cuda.h>
+#include <cuda_fp16.h>
 #include <cuda_runtime.h>
 #include <driver_types.h>
 #include <torch/extension.h>
