File size: 397 Bytes
146b945 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 | [general]
name = "adam_atan2"
universal = false
[torch]
src = [
"torch-ext/torch_binding.cpp",
"torch-ext/torch_binding.h"
]
[kernel.adam_atan2]
backend = "cuda"
cuda-capabilities = [
"7.0",
"7.2",
"7.5",
"8.0",
"8.6",
"8.7",
"8.9",
"9.0",
"10.0",
"10.1",
"11.8",
"12.0"
]
depends = ["torch", "cutlass_3_8"]
src = [
"csrc/adam_atan2.cu",
] |