medmekk
/

rmsnorm_thrust

Model card Files Files and versions Community

medmekk HF Staff commited on 9 days ago

Commit

753790d

·

verified ·

1 Parent(s): 4303459

Upload custom kernels

Files changed (1) hide show

build.toml +0 -1

build.toml CHANGED Viewed

@@ -12,4 +12,3 @@ src = [
   "rmsnorm_kernel/rmsnorm.cu",
 ]
 depends = [ "torch"]
-cuda-capabilities = [ "12.3" ]

   "rmsnorm_kernel/rmsnorm.cu",
 ]
 depends = [ "torch"]