File size: 418 Bytes
79aac9d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
[general]
version = "0.0.1"
[torch]
name = "residual_rms"
src = [
"ext-torch/registration.h",
"ext-torch/torch_binding.cpp",
"ext-torch/torch_binding.h",
]
include = ["."]
pyroot = "ext-torch"
pyext = ["py", "json"]
[kernel.residual_rms]
capabilities = ["7.0", "7.2", "7.5", "8.0", "8.6", "8.7", "8.9", "9.0"]
src = [
"residual_rms/residual_rms_dispatch.cu",
"residual_rms/compat.h",
]
depends = ["torch"]
|