chore: vendor sglang v0.5.10 snapshot
This commit is contained in:
9
third_party/sglang/sgl-kernel/tests/utils.py
vendored
Normal file
9
third_party/sglang/sgl-kernel/tests/utils.py
vendored
Normal file
@@ -0,0 +1,9 @@
|
||||
import torch
|
||||
|
||||
|
||||
def is_sm10x():
|
||||
return torch.cuda.get_device_capability() >= (10, 0)
|
||||
|
||||
|
||||
def is_hopper():
|
||||
return torch.cuda.get_device_capability() == (9, 0)
|
||||
Reference in New Issue
Block a user