adapt to sglang v0.5.2rc1 on dcu
This commit is contained in:
9
sgl-kernel/tests/utils.py
Normal file
9
sgl-kernel/tests/utils.py
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