feat: use warp reduce as a simple example (#2304)

2024-12-01 22:43:50 +08:00
parent 5c18a03733
commit 47eb139f81
7 changed files with 193 additions and 17 deletions
--- a/sgl-kernel/src/sgl-kernel/init.py
+++ b/sgl-kernel/src/sgl-kernel/init.py
@@ -0,0 +1,3 @@
+from .ops import warp_reduce
+
+__all__ = ["warp_reduce"]