Upload custom kernels
Browse files- build.toml +0 -1
build.toml
CHANGED
|
@@ -12,4 +12,3 @@ src = [
|
|
| 12 |
"rmsnorm_kernel/rmsnorm.cu",
|
| 13 |
]
|
| 14 |
depends = [ "torch"]
|
| 15 |
-
cuda-capabilities = [ "12.3" ]
|
|
|
|
| 12 |
"rmsnorm_kernel/rmsnorm.cu",
|
| 13 |
]
|
| 14 |
depends = [ "torch"]
|
|
|