Hugging Face's logo

kernels-community
/
flash-attn2

Kernels
Model card Files
xet
 Community
5
kernels-bot commited on
Commit
4279cec
·
verified ·
1 Parent(s): 5206516

Uploaded using `kernel-builder`.

Browse files
Files changed (31) hide show
  1. build/torch210-cxx11-cu126-aarch64-linux/{_flash_attn2_cuda_93b0cf2.abi3.so → _flash_attn2_cuda_85c21a0.abi3.so} +2 -2
  2. build/torch210-cxx11-cu126-aarch64-linux/_ops.py +3 -3
  3. build/torch210-cxx11-cu126-aarch64-linux/metadata.json +1 -1
  4. build/torch210-cxx11-cu128-aarch64-linux/{_flash_attn2_cuda_93b0cf2.abi3.so → _flash_attn2_cuda_85c21a0.abi3.so} +2 -2
  5. build/torch210-cxx11-cu128-aarch64-linux/_ops.py +3 -3
  6. build/torch210-cxx11-cu128-aarch64-linux/metadata.json +1 -1
  7. build/torch210-cxx11-cu130-aarch64-linux/{_flash_attn2_cuda_93b0cf2.abi3.so → _flash_attn2_cuda_85c21a0.abi3.so} +2 -2
  8. build/torch210-cxx11-cu130-aarch64-linux/_ops.py +3 -3
  9. build/torch210-cxx11-cu130-aarch64-linux/metadata.json +1 -1
  10. build/{torch212-cxx11-cu126-aarch64-linux/_flash_attn2_cuda_93b0cf2.abi3.so → torch211-cxx11-cu126-aarch64-linux/_flash_attn2_cuda_85c21a0.abi3.so} +2 -2
  11. build/torch211-cxx11-cu126-aarch64-linux/_ops.py +3 -3
  12. build/torch211-cxx11-cu126-aarch64-linux/metadata.json +1 -1
  13. build/torch211-cxx11-cu128-aarch64-linux/_flash_attn2_cuda_85c21a0.abi3.so +3 -0
  14. build/torch211-cxx11-cu128-aarch64-linux/_flash_attn2_cuda_93b0cf2.abi3.so +0 -3
  15. build/torch211-cxx11-cu128-aarch64-linux/_ops.py +3 -3
  16. build/torch211-cxx11-cu128-aarch64-linux/metadata.json +1 -1
  17. build/torch211-cxx11-cu130-aarch64-linux/_flash_attn2_cuda_85c21a0.abi3.so +3 -0
  18. build/torch211-cxx11-cu130-aarch64-linux/_flash_attn2_cuda_93b0cf2.abi3.so +0 -3
  19. build/torch211-cxx11-cu130-aarch64-linux/_ops.py +3 -3
  20. build/torch211-cxx11-cu130-aarch64-linux/metadata.json +1 -1
  21. build/{torch211-cxx11-cu126-aarch64-linux/_flash_attn2_cuda_93b0cf2.abi3.so → torch212-cxx11-cu126-aarch64-linux/_flash_attn2_cuda_85c21a0.abi3.so} +2 -2
  22. build/torch212-cxx11-cu126-aarch64-linux/_ops.py +3 -3
  23. build/torch212-cxx11-cu126-aarch64-linux/metadata.json +1 -1
  24. build/torch212-cxx11-cu130-aarch64-linux/_flash_attn2_cuda_85c21a0.abi3.so +3 -0
  25. build/torch212-cxx11-cu130-aarch64-linux/_flash_attn2_cuda_93b0cf2.abi3.so +0 -3
  26. build/torch212-cxx11-cu130-aarch64-linux/_ops.py +3 -3
  27. build/torch212-cxx11-cu130-aarch64-linux/metadata.json +1 -1
  28. build/torch212-cxx11-cu132-aarch64-linux/_flash_attn2_cuda_85c21a0.abi3.so +3 -0
  29. build/torch212-cxx11-cu132-aarch64-linux/_flash_attn2_cuda_93b0cf2.abi3.so +0 -3
  30. build/torch212-cxx11-cu132-aarch64-linux/_ops.py +3 -3
  31. build/torch212-cxx11-cu132-aarch64-linux/metadata.json +1 -1
build/torch210-cxx11-cu126-aarch64-linux/{_flash_attn2_cuda_93b0cf2.abi3.so → _flash_attn2_cuda_85c21a0.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6a2dbd72f90200c6eeb0af34b80474129ca47b9622a55b6ea387fb71cf129f8b
3
- size 448533504
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b5d9c0bbaf577980d0ec6fee304bb99b832bdab1761f89af0f045f806d9a0528
3
+ size 448608936
build/torch210-cxx11-cu126-aarch64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _flash_attn2_cuda_93b0cf2
3
- ops = torch.ops._flash_attn2_cuda_93b0cf2
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_93b0cf2::{op_name}"
 
1
  import torch
2
+ from . import _flash_attn2_cuda_85c21a0
3
+ ops = torch.ops._flash_attn2_cuda_85c21a0
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_flash_attn2_cuda_85c21a0::{op_name}"
build/torch210-cxx11-cu126-aarch64-linux/metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "name": "flash-attn2",
3
- "id": "_flash_attn2_cuda_93b0cf2",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
 
1
  {
2
  "name": "flash-attn2",
3
+ "id": "_flash_attn2_cuda_85c21a0",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
build/torch210-cxx11-cu128-aarch64-linux/{_flash_attn2_cuda_93b0cf2.abi3.so → _flash_attn2_cuda_85c21a0.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9575dbcc1f6deb66ad86fd240114761b7ca5d6c9a603d392e487afd3309bfd1a
3
- size 1037990952
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:df49147950da73dbecb34b0fe2882471704c2668112af4d0cd4c783be1e01463
3
+ size 1038067096
build/torch210-cxx11-cu128-aarch64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _flash_attn2_cuda_93b0cf2
3
- ops = torch.ops._flash_attn2_cuda_93b0cf2
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_93b0cf2::{op_name}"
 
1
  import torch
2
+ from . import _flash_attn2_cuda_85c21a0
3
+ ops = torch.ops._flash_attn2_cuda_85c21a0
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_flash_attn2_cuda_85c21a0::{op_name}"
build/torch210-cxx11-cu128-aarch64-linux/metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "name": "flash-attn2",
3
- "id": "_flash_attn2_cuda_93b0cf2",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
 
1
  {
2
  "name": "flash-attn2",
3
+ "id": "_flash_attn2_cuda_85c21a0",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
build/torch210-cxx11-cu130-aarch64-linux/{_flash_attn2_cuda_93b0cf2.abi3.so → _flash_attn2_cuda_85c21a0.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:801f0f8c09783b2cf1eca03600b3cb41b41d8af172c2b89f14e8b04d38c64c34
3
- size 1008644888
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:664859f216670f843a3e6483878a62586968be762becca2562c46ebc2a678151
3
+ size 1008655376
build/torch210-cxx11-cu130-aarch64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _flash_attn2_cuda_93b0cf2
3
- ops = torch.ops._flash_attn2_cuda_93b0cf2
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_93b0cf2::{op_name}"
 
1
  import torch
2
+ from . import _flash_attn2_cuda_85c21a0
3
+ ops = torch.ops._flash_attn2_cuda_85c21a0
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_flash_attn2_cuda_85c21a0::{op_name}"
build/torch210-cxx11-cu130-aarch64-linux/metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "name": "flash-attn2",
3
- "id": "_flash_attn2_cuda_93b0cf2",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
 
1
  {
2
  "name": "flash-attn2",
3
+ "id": "_flash_attn2_cuda_85c21a0",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
build/{torch212-cxx11-cu126-aarch64-linux/_flash_attn2_cuda_93b0cf2.abi3.so → torch211-cxx11-cu126-aarch64-linux/_flash_attn2_cuda_85c21a0.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:23515351fae9b31d35f535ba553cce63fa3ab080b0185c838103f0710522e9cc
3
- size 448533232
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d7dcc7af49adf3f3b5a4ad7d265e86be1dc5d2b33082f872166f1dba0103a836
3
+ size 448605008
build/torch211-cxx11-cu126-aarch64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _flash_attn2_cuda_93b0cf2
3
- ops = torch.ops._flash_attn2_cuda_93b0cf2
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_93b0cf2::{op_name}"
 
1
  import torch
2
+ from . import _flash_attn2_cuda_85c21a0
3
+ ops = torch.ops._flash_attn2_cuda_85c21a0
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_flash_attn2_cuda_85c21a0::{op_name}"
build/torch211-cxx11-cu126-aarch64-linux/metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "name": "flash-attn2",
3
- "id": "_flash_attn2_cuda_93b0cf2",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
 
1
  {
2
  "name": "flash-attn2",
3
+ "id": "_flash_attn2_cuda_85c21a0",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
build/torch211-cxx11-cu128-aarch64-linux/_flash_attn2_cuda_85c21a0.abi3.so ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:33ccbc5375ca903c01cd3b2720516be71dc12792fbf2f11397a1f7c1fa589038
3
+ size 1037997832
build/torch211-cxx11-cu128-aarch64-linux/_flash_attn2_cuda_93b0cf2.abi3.so DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:b87ac231a8ef6d7700ca1d6a3f92ec9a71ac703d3f78a3529c99a08947581aef
3
- size 1037987184
 
 
 
 
build/torch211-cxx11-cu128-aarch64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _flash_attn2_cuda_93b0cf2
3
- ops = torch.ops._flash_attn2_cuda_93b0cf2
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_93b0cf2::{op_name}"
 
1
  import torch
2
+ from . import _flash_attn2_cuda_85c21a0
3
+ ops = torch.ops._flash_attn2_cuda_85c21a0
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_flash_attn2_cuda_85c21a0::{op_name}"
build/torch211-cxx11-cu128-aarch64-linux/metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "name": "flash-attn2",
3
- "id": "_flash_attn2_cuda_93b0cf2",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
 
1
  {
2
  "name": "flash-attn2",
3
+ "id": "_flash_attn2_cuda_85c21a0",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
build/torch211-cxx11-cu130-aarch64-linux/_flash_attn2_cuda_85c21a0.abi3.so ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5b7f23eaf5cc0d69a1bb4b021b1401b1da1f48f4d6302d07eb7632e9d0444ea9
3
+ size 1008651464
build/torch211-cxx11-cu130-aarch64-linux/_flash_attn2_cuda_93b0cf2.abi3.so DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:9f258f1e0b43582268c0e0a56c9a7446f843730697c1e3b70bc7fd3e4ef48c86
3
- size 1008641088
 
 
 
 
build/torch211-cxx11-cu130-aarch64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _flash_attn2_cuda_93b0cf2
3
- ops = torch.ops._flash_attn2_cuda_93b0cf2
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_93b0cf2::{op_name}"
 
1
  import torch
2
+ from . import _flash_attn2_cuda_85c21a0
3
+ ops = torch.ops._flash_attn2_cuda_85c21a0
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_flash_attn2_cuda_85c21a0::{op_name}"
build/torch211-cxx11-cu130-aarch64-linux/metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "name": "flash-attn2",
3
- "id": "_flash_attn2_cuda_93b0cf2",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
 
1
  {
2
  "name": "flash-attn2",
3
+ "id": "_flash_attn2_cuda_85c21a0",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
build/{torch211-cxx11-cu126-aarch64-linux/_flash_attn2_cuda_93b0cf2.abi3.so → torch212-cxx11-cu126-aarch64-linux/_flash_attn2_cuda_85c21a0.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a97ba5d5eb2edb48237d7fa376de3d92303eabdb0f45e678f9807c89bcd6835a
3
- size 448529832
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fc6154096b681b4a88167b3d0a3822b56460a047e87cd178c5b7160e33ad0929
3
+ size 448533000
build/torch212-cxx11-cu126-aarch64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _flash_attn2_cuda_93b0cf2
3
- ops = torch.ops._flash_attn2_cuda_93b0cf2
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_93b0cf2::{op_name}"
 
1
  import torch
2
+ from . import _flash_attn2_cuda_85c21a0
3
+ ops = torch.ops._flash_attn2_cuda_85c21a0
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_flash_attn2_cuda_85c21a0::{op_name}"
build/torch212-cxx11-cu126-aarch64-linux/metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "name": "flash-attn2",
3
- "id": "_flash_attn2_cuda_93b0cf2",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
 
1
  {
2
  "name": "flash-attn2",
3
+ "id": "_flash_attn2_cuda_85c21a0",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
build/torch212-cxx11-cu130-aarch64-linux/_flash_attn2_cuda_85c21a0.abi3.so ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8f799b26743dfa81f06076108d96f800c5778f57d3561c261a5b940682129604
3
+ size 1008645144
build/torch212-cxx11-cu130-aarch64-linux/_flash_attn2_cuda_93b0cf2.abi3.so DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:1008b71ff56f162d3b920c32bba353fc8533baf4a895ef5345e995bc149d3cb5
3
- size 1008644656
 
 
 
 
build/torch212-cxx11-cu130-aarch64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _flash_attn2_cuda_93b0cf2
3
- ops = torch.ops._flash_attn2_cuda_93b0cf2
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_93b0cf2::{op_name}"
 
1
  import torch
2
+ from . import _flash_attn2_cuda_85c21a0
3
+ ops = torch.ops._flash_attn2_cuda_85c21a0
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_flash_attn2_cuda_85c21a0::{op_name}"
build/torch212-cxx11-cu130-aarch64-linux/metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "name": "flash-attn2",
3
- "id": "_flash_attn2_cuda_93b0cf2",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
 
1
  {
2
  "name": "flash-attn2",
3
+ "id": "_flash_attn2_cuda_85c21a0",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
build/torch212-cxx11-cu132-aarch64-linux/_flash_attn2_cuda_85c21a0.abi3.so ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:783f89cd295676fd01ea025a5caf3ab14df2752cd42ace7b29fa9ae6cf4526af
3
+ size 1026021240
build/torch212-cxx11-cu132-aarch64-linux/_flash_attn2_cuda_93b0cf2.abi3.so DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:f3c04a8219e405c29f2afc41fa0a313cdb7374d9e30e44fb03e24494a128684c
3
- size 1026020752
 
 
 
 
build/torch212-cxx11-cu132-aarch64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _flash_attn2_cuda_93b0cf2
3
- ops = torch.ops._flash_attn2_cuda_93b0cf2
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_93b0cf2::{op_name}"
 
1
  import torch
2
+ from . import _flash_attn2_cuda_85c21a0
3
+ ops = torch.ops._flash_attn2_cuda_85c21a0
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_flash_attn2_cuda_85c21a0::{op_name}"
build/torch212-cxx11-cu132-aarch64-linux/metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "name": "flash-attn2",
3
- "id": "_flash_attn2_cuda_93b0cf2",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
 
1
  {
2
  "name": "flash-attn2",
3
+ "id": "_flash_attn2_cuda_85c21a0",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],