kernels-community
/

triton_kernels

Model card Files Files and versions

triton_kernels / torch-ext

400 kB

6 contributors

History: 13 commits

janantos's picture

Update torch-ext/triton_kernels/target_info.py num_sms() to support pytorch-rocm cuda and gpt-oss models

8225dcf verified about 1 month ago

triton_kernels
Update torch-ext/triton_kernels/target_info.py num_sms() to support pytorch-rocm cuda and gpt-oss models about 1 month ago