[Feature] CUDA Green Context Support (#7649)

2025-07-15 02:49:16 +08:00
parent d4d0c7c367
commit 1ebec1a8b0
9 changed files with 190 additions and 0 deletions
--- a/sgl-kernel/python/sgl_kernel/init.py
+++ b/sgl-kernel/python/sgl_kernel/init.py
@@ -81,6 +81,7 @@ from sgl_kernel.sampling import (
    top_p_renorm_prob,
    top_p_sampling_from_probs,
 )
+from sgl_kernel.spatial import create_greenctx_stream_by_value, get_sm_available
 from sgl_kernel.speculative import (
    build_tree_kernel_efficient,
    segment_packbits,