Hugging Face's logo

Kernels:
kernels-community
/
flash-attn2
Trusted publisher

Kernel card Files
xet
 Community
kernels-bot commited on
Commit
9bc9eea
·
verified ·
1 Parent(s): 1779e8b

Uploaded using `kernel-builder`.

Browse files
Files changed (17) hide show
  1. build/torch210-cxx11-cu126-aarch64-linux/{_flash_attn2_cuda_cbe520f.abi3.so → _flash_attn2_cuda_958fa11.abi3.so} +2 -2
  2. build/torch210-cxx11-cu126-aarch64-linux/_ops.py +3 -3
  3. build/torch210-cxx11-cu128-aarch64-linux/{_flash_attn2_cuda_cbe520f.abi3.so → _flash_attn2_cuda_958fa11.abi3.so} +2 -2
  4. build/torch210-cxx11-cu128-aarch64-linux/_ops.py +3 -3
  5. build/torch210-cxx11-cu130-aarch64-linux/{_flash_attn2_cuda_cbe520f.abi3.so → _flash_attn2_cuda_958fa11.abi3.so} +2 -2
  6. build/torch210-cxx11-cu130-aarch64-linux/_ops.py +3 -3
  7. build/torch211-cxx11-cu126-aarch64-linux/{_flash_attn2_cuda_cbe520f.abi3.so → _flash_attn2_cuda_958fa11.abi3.so} +2 -2
  8. build/torch211-cxx11-cu126-aarch64-linux/_ops.py +3 -3
  9. build/torch211-cxx11-cu128-aarch64-linux/_flash_attn2_cuda_958fa11.abi3.so +3 -0
  10. build/torch211-cxx11-cu128-aarch64-linux/_flash_attn2_cuda_cbe520f.abi3.so +0 -3
  11. build/torch211-cxx11-cu128-aarch64-linux/_ops.py +3 -3
  12. build/torch211-cxx11-cu130-aarch64-linux/_flash_attn2_cuda_958fa11.abi3.so +3 -0
  13. build/torch211-cxx11-cu130-aarch64-linux/_flash_attn2_cuda_cbe520f.abi3.so +0 -3
  14. build/torch211-cxx11-cu130-aarch64-linux/_ops.py +3 -3
  15. build/torch29-cxx11-cu129-aarch64-linux/_flash_attn2_cuda_958fa11.abi3.so +3 -0
  16. build/torch29-cxx11-cu129-aarch64-linux/_flash_attn2_cuda_cbe520f.abi3.so +0 -3
  17. build/torch29-cxx11-cu129-aarch64-linux/_ops.py +3 -3
build/torch210-cxx11-cu126-aarch64-linux/{_flash_attn2_cuda_cbe520f.abi3.so → _flash_attn2_cuda_958fa11.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4aadf18e4286f334848803a2f193b940a5e8deb5d8d92c6629e6c62c4e6e0fb4
3
- size 448533496
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6fc8bd519ff37c3e79c24bd6bd252352b01f070fa8f7c1a8432c45f2bc22b52f
3
+ size 448533504
build/torch210-cxx11-cu126-aarch64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _flash_attn2_cuda_cbe520f
3
- ops = torch.ops._flash_attn2_cuda_cbe520f
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_cbe520f::{op_name}"
 
1
  import torch
2
+ from . import _flash_attn2_cuda_958fa11
3
+ ops = torch.ops._flash_attn2_cuda_958fa11
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_flash_attn2_cuda_958fa11::{op_name}"
build/torch210-cxx11-cu128-aarch64-linux/{_flash_attn2_cuda_cbe520f.abi3.so → _flash_attn2_cuda_958fa11.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7fd05be2d6ca2b56189bbecd2c15d94fc2324fed830d68a4b0ce7197301a4efe
3
- size 1037990944
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:97e690df2995d1d87a4cc3f8eb17f0d3ef80fc36796ef6d593f13c7e6aedf3fd
3
+ size 1037990952
build/torch210-cxx11-cu128-aarch64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _flash_attn2_cuda_cbe520f
3
- ops = torch.ops._flash_attn2_cuda_cbe520f
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_cbe520f::{op_name}"
 
1
  import torch
2
+ from . import _flash_attn2_cuda_958fa11
3
+ ops = torch.ops._flash_attn2_cuda_958fa11
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_flash_attn2_cuda_958fa11::{op_name}"
build/torch210-cxx11-cu130-aarch64-linux/{_flash_attn2_cuda_cbe520f.abi3.so → _flash_attn2_cuda_958fa11.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6addea18d7b47d9a8a875159619dce75828d68a9fd7b7ddafc7c5d0c275bdd71
3
- size 1008644344
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:26dbdabe7ffa5834612cf37913f637c91aa991d757f4e42b4876c1921ecbeeb7
3
+ size 1008644888
build/torch210-cxx11-cu130-aarch64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _flash_attn2_cuda_cbe520f
3
- ops = torch.ops._flash_attn2_cuda_cbe520f
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_cbe520f::{op_name}"
 
1
  import torch
2
+ from . import _flash_attn2_cuda_958fa11
3
+ ops = torch.ops._flash_attn2_cuda_958fa11
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_flash_attn2_cuda_958fa11::{op_name}"
build/torch211-cxx11-cu126-aarch64-linux/{_flash_attn2_cuda_cbe520f.abi3.so → _flash_attn2_cuda_958fa11.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5535583b5c8cc95df01fe46f9bc0ce1dc43993b0eb3a2cf23ca0cc42063a6afe
3
- size 448529824
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4c35ba9816d590f803a80a45fcb5be581dfceb15af4f86daf3758ba46f56ef3b
3
+ size 448529832
build/torch211-cxx11-cu126-aarch64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _flash_attn2_cuda_cbe520f
3
- ops = torch.ops._flash_attn2_cuda_cbe520f
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_cbe520f::{op_name}"
 
1
  import torch
2
+ from . import _flash_attn2_cuda_958fa11
3
+ ops = torch.ops._flash_attn2_cuda_958fa11
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_flash_attn2_cuda_958fa11::{op_name}"
build/torch211-cxx11-cu128-aarch64-linux/_flash_attn2_cuda_958fa11.abi3.so ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cef2b4d449b3efb635e6226e2d0257b080677daa631e732c73905f94ea9bdb7c
3
+ size 1037987184
build/torch211-cxx11-cu128-aarch64-linux/_flash_attn2_cuda_cbe520f.abi3.so DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:f870c03a0bd791f473c89012df5401b96f1bbba34e5bd18a99d36aec5318ab29
3
- size 1037987168
 
 
 
 
build/torch211-cxx11-cu128-aarch64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _flash_attn2_cuda_cbe520f
3
- ops = torch.ops._flash_attn2_cuda_cbe520f
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_cbe520f::{op_name}"
 
1
  import torch
2
+ from . import _flash_attn2_cuda_958fa11
3
+ ops = torch.ops._flash_attn2_cuda_958fa11
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_flash_attn2_cuda_958fa11::{op_name}"
build/torch211-cxx11-cu130-aarch64-linux/_flash_attn2_cuda_958fa11.abi3.so ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:27fa941b83482a166cbd246e95694161e89557eef2a5b284016d3e4765dce49a
3
+ size 1008641088
build/torch211-cxx11-cu130-aarch64-linux/_flash_attn2_cuda_cbe520f.abi3.so DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:cdc0e35227f5090374c5a1a301852bc06da0c5115721709534d9ae999e6c2ae8
3
- size 1008640568
 
 
 
 
build/torch211-cxx11-cu130-aarch64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _flash_attn2_cuda_cbe520f
3
- ops = torch.ops._flash_attn2_cuda_cbe520f
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_cbe520f::{op_name}"
 
1
  import torch
2
+ from . import _flash_attn2_cuda_958fa11
3
+ ops = torch.ops._flash_attn2_cuda_958fa11
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_flash_attn2_cuda_958fa11::{op_name}"
build/torch29-cxx11-cu129-aarch64-linux/_flash_attn2_cuda_958fa11.abi3.so ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b3363c49d10c61078a9a05359fef626f4ed3b0ee0f68eb5a88828823bfcfe651
3
+ size 1043164184
build/torch29-cxx11-cu129-aarch64-linux/_flash_attn2_cuda_cbe520f.abi3.so DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:8575411928c4ee7d0066d833de0e2f7463214f1c8fc564dac52f92780a2312d2
3
- size 1043164168
 
 
 
 
build/torch29-cxx11-cu129-aarch64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _flash_attn2_cuda_cbe520f
3
- ops = torch.ops._flash_attn2_cuda_cbe520f
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_cbe520f::{op_name}"
 
1
  import torch
2
+ from . import _flash_attn2_cuda_958fa11
3
+ ops = torch.ops._flash_attn2_cuda_958fa11
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_flash_attn2_cuda_958fa11::{op_name}"