Hugging Face's logo

kernels-community
/
flash-attn2

Kernels
Model card Files
xet
 Community
5
kernels-bot commited on
Commit
ec2b003
·
verified ·
1 Parent(s): 97749c6

Uploaded using `kernel-builder`.

Browse files
Files changed (27) hide show
  1. build/torch210-cxx11-cu126-aarch64-linux/{_flash_attn2_cuda_4754244.abi3.so → _flash_attn2_cuda_e7ab8ac.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_4754244.abi3.so → _flash_attn2_cuda_e7ab8ac.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_4754244.abi3.so → _flash_attn2_cuda_e7ab8ac.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_4754244.abi3.so → _flash_attn2_cuda_e7ab8ac.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_4754244.abi3.so → _flash_attn2_cuda_e7ab8ac.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_4754244.abi3.so → _flash_attn2_cuda_e7ab8ac.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_4754244.abi3.so → _flash_attn2_cuda_e7ab8ac.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_4754244.abi3.so → _flash_attn2_cuda_e7ab8ac.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_4754244.abi3.so → _flash_attn2_cuda_e7ab8ac.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_4754244.abi3.so → _flash_attn2_cuda_e7ab8ac.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:15afeb2497470a5595fe951494b500d22edd06c4b98d4497bab686934f7d8f17
3
  size 448608936
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7fd3c74f839515b5bd041915cc7e2480a87cdf81e6a9784b2589c5bd7a77365a
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_4754244
3
- ops = torch.ops._flash_attn2_cuda_4754244
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_4754244::{op_name}"
 
1
  import torch
2
+ from . import _flash_attn2_cuda_e7ab8ac
3
+ ops = torch.ops._flash_attn2_cuda_e7ab8ac
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_flash_attn2_cuda_e7ab8ac::{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_4754244",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
 
1
  {
2
  "name": "flash-attn2",
3
+ "id": "_flash_attn2_cuda_e7ab8ac",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
build/torch210-cxx11-cu128-aarch64-linux/{_flash_attn2_cuda_4754244.abi3.so → _flash_attn2_cuda_e7ab8ac.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:67276854d1b19d1f24860439848391465878ef30cbd2e97712082541b5509361
3
  size 1038067096
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1c5c1e481903124642e59694fcf37ce2a52c37523040e95b7f872cff85c2f385
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_4754244
3
- ops = torch.ops._flash_attn2_cuda_4754244
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_4754244::{op_name}"
 
1
  import torch
2
+ from . import _flash_attn2_cuda_e7ab8ac
3
+ ops = torch.ops._flash_attn2_cuda_e7ab8ac
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_flash_attn2_cuda_e7ab8ac::{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_4754244",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
 
1
  {
2
  "name": "flash-attn2",
3
+ "id": "_flash_attn2_cuda_e7ab8ac",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
build/torch210-cxx11-cu130-aarch64-linux/{_flash_attn2_cuda_4754244.abi3.so → _flash_attn2_cuda_e7ab8ac.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:fdeb97904328a64e7b03551f865e45d87f17b941ba26ab4651a62886e62b070e
3
  size 1008655376
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3c173db33d2ba4917f4c10da9c09ae5be1032c82780422069d4d6881d520f853
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_4754244
3
- ops = torch.ops._flash_attn2_cuda_4754244
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_4754244::{op_name}"
 
1
  import torch
2
+ from . import _flash_attn2_cuda_e7ab8ac
3
+ ops = torch.ops._flash_attn2_cuda_e7ab8ac
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_flash_attn2_cuda_e7ab8ac::{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_4754244",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
 
1
  {
2
  "name": "flash-attn2",
3
+ "id": "_flash_attn2_cuda_e7ab8ac",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
build/torch211-cxx11-cu126-aarch64-linux/{_flash_attn2_cuda_4754244.abi3.so → _flash_attn2_cuda_e7ab8ac.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:aeccf5649dabd7f66e68ca8c38daf0d91ff6e29a03732f1b671617e70420b1f4
3
  size 448605008
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:95768711b37730bcd41473f5280444c818cfd3b2427d96172d625e4cf6270fc1
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_4754244
3
- ops = torch.ops._flash_attn2_cuda_4754244
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_4754244::{op_name}"
 
1
  import torch
2
+ from . import _flash_attn2_cuda_e7ab8ac
3
+ ops = torch.ops._flash_attn2_cuda_e7ab8ac
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_flash_attn2_cuda_e7ab8ac::{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_4754244",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
 
1
  {
2
  "name": "flash-attn2",
3
+ "id": "_flash_attn2_cuda_e7ab8ac",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
build/torch211-cxx11-cu128-aarch64-linux/{_flash_attn2_cuda_4754244.abi3.so → _flash_attn2_cuda_e7ab8ac.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:05dd5200ba122ac8406262480b253d58033b50b950272be36387993197841e7e
3
  size 1037997832
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aad1ba9237d08f6c8fbbcf8cb43ace019b92ebfcf9fe8638a34956d32dce648a
3
  size 1037997832
build/torch211-cxx11-cu128-aarch64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _flash_attn2_cuda_4754244
3
- ops = torch.ops._flash_attn2_cuda_4754244
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_4754244::{op_name}"
 
1
  import torch
2
+ from . import _flash_attn2_cuda_e7ab8ac
3
+ ops = torch.ops._flash_attn2_cuda_e7ab8ac
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_flash_attn2_cuda_e7ab8ac::{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_4754244",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
 
1
  {
2
  "name": "flash-attn2",
3
+ "id": "_flash_attn2_cuda_e7ab8ac",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
build/torch211-cxx11-cu130-aarch64-linux/{_flash_attn2_cuda_4754244.abi3.so → _flash_attn2_cuda_e7ab8ac.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:624afe94e3b615d2fd1df94655db7538e2d34d02878c97a8dcff686cedf3f879
3
  size 1008651464
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6089e0ebceba3a3192ac8e0805d5076a51c2b4bfd6ea315a06e910171c67decd
3
  size 1008651464
build/torch211-cxx11-cu130-aarch64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _flash_attn2_cuda_4754244
3
- ops = torch.ops._flash_attn2_cuda_4754244
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_4754244::{op_name}"
 
1
  import torch
2
+ from . import _flash_attn2_cuda_e7ab8ac
3
+ ops = torch.ops._flash_attn2_cuda_e7ab8ac
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_flash_attn2_cuda_e7ab8ac::{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_4754244",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
 
1
  {
2
  "name": "flash-attn2",
3
+ "id": "_flash_attn2_cuda_e7ab8ac",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
build/torch212-cxx11-cu126-aarch64-linux/{_flash_attn2_cuda_4754244.abi3.so → _flash_attn2_cuda_e7ab8ac.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e59759e2e5cdddace15638920af5a9c6a8613869d1b27996542475f405b2f078
3
  size 448533000
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:10a5f75163f6672f2eefee05bf500b1d249aa22161d5c46c6ea823014bd446ec
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_4754244
3
- ops = torch.ops._flash_attn2_cuda_4754244
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_4754244::{op_name}"
 
1
  import torch
2
+ from . import _flash_attn2_cuda_e7ab8ac
3
+ ops = torch.ops._flash_attn2_cuda_e7ab8ac
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_flash_attn2_cuda_e7ab8ac::{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_4754244",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
 
1
  {
2
  "name": "flash-attn2",
3
+ "id": "_flash_attn2_cuda_e7ab8ac",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
build/torch212-cxx11-cu130-aarch64-linux/{_flash_attn2_cuda_4754244.abi3.so → _flash_attn2_cuda_e7ab8ac.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c5dcdf7a1980ed4ee6f64754d9229f6743f8b0e56c8b628583a5812d2510635e
3
  size 1008645144
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7bfe075b493ec678e7fa26312512bd5805e5beda98e052740d40c42ce484bfe8
3
  size 1008645144
build/torch212-cxx11-cu130-aarch64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _flash_attn2_cuda_4754244
3
- ops = torch.ops._flash_attn2_cuda_4754244
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_4754244::{op_name}"
 
1
  import torch
2
+ from . import _flash_attn2_cuda_e7ab8ac
3
+ ops = torch.ops._flash_attn2_cuda_e7ab8ac
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_flash_attn2_cuda_e7ab8ac::{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_4754244",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
 
1
  {
2
  "name": "flash-attn2",
3
+ "id": "_flash_attn2_cuda_e7ab8ac",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
build/torch212-cxx11-cu132-aarch64-linux/{_flash_attn2_cuda_4754244.abi3.so → _flash_attn2_cuda_e7ab8ac.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:93e1b9e6bea8f1a3d793b2d313a6d7171295eaddd8a7099e33b36dee7a8b9716
3
  size 1026021240
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:47fc28b5b6723497f6f39e5ca42db842b4e85fe75d6f6b050641091e72e48db6
3
  size 1026021240
build/torch212-cxx11-cu132-aarch64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _flash_attn2_cuda_4754244
3
- ops = torch.ops._flash_attn2_cuda_4754244
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_flash_attn2_cuda_4754244::{op_name}"
 
1
  import torch
2
+ from . import _flash_attn2_cuda_e7ab8ac
3
+ ops = torch.ops._flash_attn2_cuda_e7ab8ac
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_flash_attn2_cuda_e7ab8ac::{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_4754244",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],
 
1
  {
2
  "name": "flash-attn2",
3
+ "id": "_flash_attn2_cuda_e7ab8ac",
4
  "version": 1,
5
  "license": "BSD-3-Clause",
6
  "python-depends": [],