mirror of
https://github.com/kvcache-ai/sglang.git
synced 2026-06-30 19:57:52 +00:00
10 lines
163 B
Python
10 lines
163 B
Python
import torch
|
|
|
|
|
|
def is_sm10x():
|
|
return torch.cuda.get_device_capability() >= (10, 0)
|
|
|
|
|
|
def is_hopper():
|
|
return torch.cuda.get_device_capability() == (9, 0)
|