sglang_v0.5.2/sglang/sgl-kernel/tests/utils.py

import torch


def is_sm10x():
    return torch.cuda.get_device_capability() >= (10, 0)


def is_hopper():
    return torch.cuda.get_device_capability() == (9, 0)