File size: 345 Bytes
e52d1ec b865e0f e52d1ec b865e0f e52d1ec | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | [general]
name = "get_position_ids"
[torch]
src = [
"ext-torch/torch_binding.cpp",
"ext-torch/torch_binding.h",
]
include = ["."]
pyext = ["py", "json"]
[kernel.get_position_ids]
cuda-capabilities = ["7.0", "7.2", "7.5", "8.0", "8.6", "8.7", "8.9", "9.0"]
src = ["get_position_ids/get_position_ids.cu"]
include = ["."]
depends = ["torch"]
|