Hugging Face's logo

kernels-community
/
flash-attn2

Kernels
Model card Files
xet
 Community
5
kernels-bot commited on
Commit
74ea86d
·
verified ·
1 Parent(s): 60f6020

Uploaded using `kernel-builder`.

Browse files
Files changed (27) hide show
  1. build/torch210-cxx11-cu126-aarch64-linux/{_flash_attn2_cuda_d36888d.abi3.so → _flash_attn2_cuda_93b0cf2.abi3.so} +1 -1
  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_d36888d.abi3.so → _flash_attn2_cuda_93b0cf2.abi3.so} +1 -1
  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_d36888d.abi3.so → _flash_attn2_cuda_93b0cf2.abi3.so} +1 -1
  8. build/torch210-cxx11-cu130-aarch64-linux/_ops.py +3 -3
  9. build/torch210-cxx11-cu130-aarch64-linux/metadata.json +1 -1
  10. build/torch211-cxx11-cu126-aarch64-linux/{_flash_attn2_cuda_d36888d.abi3.so → _flash_attn2_cuda_93b0cf2.abi3.so} +1 -1
  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_d36888d.abi3.so → _flash_attn2_cuda_93b0cf2.abi3.so} +1 -1
  14. build/torch211-cxx11-cu128-aarch64-linux/_ops.py +3 -3
  15. build/torch211-cxx11-cu128-aarch64-linux/metadata.json +1 -1
  16. build/torch211-cxx11-cu130-aarch64-linux/{_flash_attn2_cuda_d36888d.abi3.so → _flash_attn2_cuda_93b0cf2.abi3.so} +1 -1
  17. build/torch211-cxx11-cu130-aarch64-linux/_ops.py +3 -3
  18. build/torch211-cxx11-cu130-aarch64-linux/metadata.json +1 -1
  19. build/torch212-cxx11-cu126-aarch64-linux/{_flash_attn2_cuda_d36888d.abi3.so → _flash_attn2_cuda_93b0cf2.abi3.so} +1 -1
  20. build/torch212-cxx11-cu126-aarch64-linux/_ops.py +3 -3
  21. build/torch212-cxx11-cu126-aarch64-linux/metadata.json +1 -1
  22. build/torch212-cxx11-cu130-aarch64-linux/{_flash_attn2_cuda_d36888d.abi3.so → _flash_attn2_cuda_93b0cf2.abi3.so} +1 -1
  23. build/torch212-cxx11-cu130-aarch64-linux/_ops.py +3 -3
  24. build/torch212-cxx11-cu130-aarch64-linux/metadata.json +1 -1
  25. build/torch212-cxx11-cu132-aarch64-linux/{_flash_attn2_cuda_d36888d.abi3.so → _flash_attn2_cuda_93b0cf2.abi3.so} +1 -1
  26. build/torch212-cxx11-cu132-aarch64-linux/_ops.py +3 -3
  27. build/torch212-cxx11-cu132-aarch64-linux/metadata.json +1 -1
build/torch210-cxx11-cu126-aarch64-linux/{_flash_attn2_cuda_d36888d.abi3.so → _flash_attn2_cuda_93b0cf2.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:210d9aa4af157da8f90e3fa3726709ff15a8871323c492299fef11024866d31e
3
  size 448533504
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6a2dbd72f90200c6eeb0af34b80474129ca47b9622a55b6ea387fb71cf129f8b
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_d36888d
3
- ops = torch.ops._flash_attn2_cuda_d36888d
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_d36888d::{op_name}"
 
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}"
build/torch210-cxx11-cu126-aarch64-linux/metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "name": "flash-attn2",
3
- "id": "_flash_attn2_cuda_d36888d",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
 
1
  {
2
  "name": "flash-attn2",
3
+ "id": "_flash_attn2_cuda_93b0cf2",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
build/torch210-cxx11-cu128-aarch64-linux/{_flash_attn2_cuda_d36888d.abi3.so → _flash_attn2_cuda_93b0cf2.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ad0c23d70237f75a307d586e77fc692a374b0002b4270bcb54e9610e54e654db
3
  size 1037990952
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9575dbcc1f6deb66ad86fd240114761b7ca5d6c9a603d392e487afd3309bfd1a
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_d36888d
3
- ops = torch.ops._flash_attn2_cuda_d36888d
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_d36888d::{op_name}"
 
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}"
build/torch210-cxx11-cu128-aarch64-linux/metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "name": "flash-attn2",
3
- "id": "_flash_attn2_cuda_d36888d",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
 
1
  {
2
  "name": "flash-attn2",
3
+ "id": "_flash_attn2_cuda_93b0cf2",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
build/torch210-cxx11-cu130-aarch64-linux/{_flash_attn2_cuda_d36888d.abi3.so → _flash_attn2_cuda_93b0cf2.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:bdfadee605aec6a9d18b5c1611e4bd159cc34dba9fccf9466cfb056e3d5c7e20
3
  size 1008644888
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:801f0f8c09783b2cf1eca03600b3cb41b41d8af172c2b89f14e8b04d38c64c34
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_d36888d
3
- ops = torch.ops._flash_attn2_cuda_d36888d
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_d36888d::{op_name}"
 
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}"
build/torch210-cxx11-cu130-aarch64-linux/metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "name": "flash-attn2",
3
- "id": "_flash_attn2_cuda_d36888d",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
 
1
  {
2
  "name": "flash-attn2",
3
+ "id": "_flash_attn2_cuda_93b0cf2",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
build/torch211-cxx11-cu126-aarch64-linux/{_flash_attn2_cuda_d36888d.abi3.so → _flash_attn2_cuda_93b0cf2.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c473a70079c31a343dfe4778b2446722be65d86855a97cfcb700d6bf98d34dc8
3
  size 448529832
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a97ba5d5eb2edb48237d7fa376de3d92303eabdb0f45e678f9807c89bcd6835a
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_d36888d
3
- ops = torch.ops._flash_attn2_cuda_d36888d
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_d36888d::{op_name}"
 
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}"
build/torch211-cxx11-cu126-aarch64-linux/metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "name": "flash-attn2",
3
- "id": "_flash_attn2_cuda_d36888d",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
 
1
  {
2
  "name": "flash-attn2",
3
+ "id": "_flash_attn2_cuda_93b0cf2",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
build/torch211-cxx11-cu128-aarch64-linux/{_flash_attn2_cuda_d36888d.abi3.so → _flash_attn2_cuda_93b0cf2.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:19318bfdc9dcd37aff92bca3572a192f5f5be58c61615b20430a720041876bcb
3
  size 1037987184
 
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_d36888d
3
- ops = torch.ops._flash_attn2_cuda_d36888d
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_d36888d::{op_name}"
 
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}"
build/torch211-cxx11-cu128-aarch64-linux/metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "name": "flash-attn2",
3
- "id": "_flash_attn2_cuda_d36888d",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
 
1
  {
2
  "name": "flash-attn2",
3
+ "id": "_flash_attn2_cuda_93b0cf2",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
build/torch211-cxx11-cu130-aarch64-linux/{_flash_attn2_cuda_d36888d.abi3.so → _flash_attn2_cuda_93b0cf2.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0664f1c07a67ae96b270ec33cb4d873c5f00c649aec849badec089eb84ad8c23
3
  size 1008641088
 
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_d36888d
3
- ops = torch.ops._flash_attn2_cuda_d36888d
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_d36888d::{op_name}"
 
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}"
build/torch211-cxx11-cu130-aarch64-linux/metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "name": "flash-attn2",
3
- "id": "_flash_attn2_cuda_d36888d",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
 
1
  {
2
  "name": "flash-attn2",
3
+ "id": "_flash_attn2_cuda_93b0cf2",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
build/torch212-cxx11-cu126-aarch64-linux/{_flash_attn2_cuda_d36888d.abi3.so → _flash_attn2_cuda_93b0cf2.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:32fa917d28a8a7a26f3a566a405f4eb10a2ec5fb760f92967c4f348def9b0282
3
  size 448533232
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:23515351fae9b31d35f535ba553cce63fa3ab080b0185c838103f0710522e9cc
3
  size 448533232
build/torch212-cxx11-cu126-aarch64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _flash_attn2_cuda_d36888d
3
- ops = torch.ops._flash_attn2_cuda_d36888d
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_d36888d::{op_name}"
 
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}"
build/torch212-cxx11-cu126-aarch64-linux/metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "name": "flash-attn2",
3
- "id": "_flash_attn2_cuda_d36888d",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
 
1
  {
2
  "name": "flash-attn2",
3
+ "id": "_flash_attn2_cuda_93b0cf2",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
build/torch212-cxx11-cu130-aarch64-linux/{_flash_attn2_cuda_d36888d.abi3.so → _flash_attn2_cuda_93b0cf2.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0e6f25845bb4764232f14fd738d19937f7067183bdda69dd10f6cf9f48e20618
3
  size 1008644656
 
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_d36888d
3
- ops = torch.ops._flash_attn2_cuda_d36888d
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_d36888d::{op_name}"
 
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}"
build/torch212-cxx11-cu130-aarch64-linux/metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "name": "flash-attn2",
3
- "id": "_flash_attn2_cuda_d36888d",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
 
1
  {
2
  "name": "flash-attn2",
3
+ "id": "_flash_attn2_cuda_93b0cf2",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
build/torch212-cxx11-cu132-aarch64-linux/{_flash_attn2_cuda_d36888d.abi3.so → _flash_attn2_cuda_93b0cf2.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:02c03bfd1771c1ff9ff6bc16cc2485d87ad3d83946daf43576d6318c858a8a17
3
  size 1026020752
 
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_d36888d
3
- ops = torch.ops._flash_attn2_cuda_d36888d
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_d36888d::{op_name}"
 
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}"
build/torch212-cxx11-cu132-aarch64-linux/metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "name": "flash-attn2",
3
- "id": "_flash_attn2_cuda_d36888d",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
 
1
  {
2
  "name": "flash-attn2",
3
+ "id": "_flash_attn2_cuda_93b0cf2",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],