misc: update build setup (#2306)

2024-12-02 02:03:49 +08:00
parent 7301a39b13
commit 28bc60dcab
5 changed files with 46 additions and 1 deletions
--- a/sgl-kernel/src/sgl-kernel/csrc/warp_reduce.cc
+++ b/sgl-kernel/src/sgl-kernel/csrc/warp_reduce.cc
@@ -1,5 +1,4 @@
 #include <torch/extension.h>
-#include <vector>

 torch::Tensor warp_reduce_cuda(torch::Tensor input);