Kernels
triton_kernels / torch-ext
400 kB
janantos's picture
Update torch-ext/triton_kernels/target_info.py num_sms() to support pytorch-rocm cuda and gpt-oss models
8225dcf verified