kernels-community
/

paged-attention

danieldk HF Staff commited on 19 days ago

Commit

3f98f45

1 Parent(s): cea0337

Sync capabilities with upstream

Files changed (1) hide show

build.toml CHANGED Viewed

@@ -8,7 +8,7 @@ src = [
 ]
 [kernel.cuda_utils]
-cuda-capabilities = [ "7.0", "7.2", "7.5", "8.0", "8.6", "8.7", "8.9", "9.0" ]
 src = [
   "cuda-utils/cuda_utils_kernels.cu",
 ]
@@ -16,7 +16,7 @@ depends = []
 [kernel.paged_attention]
-cuda-capabilities = [ "7.0", "7.2", "7.5", "8.0", "8.6", "8.7", "8.9", "9.0" ]
 src = [
   "paged-attention/attention/attention_dtypes.h",
   "paged-attention/attention/attention_generic.cuh",

 ]
 [kernel.cuda_utils]
+cuda-capabilities = ["7.0", "7.2", "7.5", "8.0", "8.6", "8.7", "8.9", "9.0", "10.0", "10.1", "12.0"]
 src = [
   "cuda-utils/cuda_utils_kernels.cu",
 ]
 [kernel.paged_attention]
+cuda-capabilities = ["7.0", "7.2", "7.5", "8.0", "8.6", "8.7", "8.9", "9.0", "10.0", "10.1", "12.0"]
 src = [
   "paged-attention/attention/attention_dtypes.h",
   "paged-attention/attention/attention_generic.cuh",