Set CUDA capabilities for bgmv
Browse files- build.toml +1 -0
build.toml
CHANGED
|
@@ -32,6 +32,7 @@ depends = [ "torch" ]
|
|
| 32 |
|
| 33 |
[kernel.bgmv]
|
| 34 |
language = "cuda"
|
|
|
|
| 35 |
src = [
|
| 36 |
"bgmv/bgmv_all.cu",
|
| 37 |
"bgmv/bgmv_impl.cuh",
|
|
|
|
| 32 |
|
| 33 |
[kernel.bgmv]
|
| 34 |
language = "cuda"
|
| 35 |
+
cuda-capabilities = [ "8.0", "8.6", "8.7", "8.9", "9.0", "10.0", "10.1", "12.0" ]
|
| 36 |
src = [
|
| 37 |
"bgmv/bgmv_all.cu",
|
| 38 |
"bgmv/bgmv_impl.cuh",
|