Promote latest kernel artifacts to main
Browse files- README.md +0 -9
- build/torch212-cxx11-cu130-x86_64-linux/_ops.py +3 -3
- build/{torch213-cxx11-cu130-x86_64-linux/_sageattention2_blackwell_cuda_5a0fd7c.abi3.so → torch212-cxx11-cu130-x86_64-linux/_sageattention2_blackwell_cuda_5542189.abi3.so} +2 -2
- build/torch212-cxx11-cu130-x86_64-linux/metadata.json +6 -6
- build/torch212-cxx11-cu130-x86_64-linux/sageattention2_blackwell/__init__.py +0 -26
- build/torch212-cxx11-cu132-x86_64-linux/_ops.py +3 -3
- build/{torch212-cxx11-cu130-x86_64-linux/_sageattention2_blackwell_cuda_5a0fd7c.abi3.so → torch212-cxx11-cu132-x86_64-linux/_sageattention2_blackwell_cuda_5542189.abi3.so} +2 -2
- build/torch212-cxx11-cu132-x86_64-linux/metadata.json +6 -6
- build/torch212-cxx11-cu132-x86_64-linux/sageattention2_blackwell/__init__.py +0 -26
- build/torch213-cxx11-cu130-x86_64-linux/_ops.py +3 -3
- build/{torch213-cxx11-cu132-x86_64-linux/_sageattention2_blackwell_cuda_5a0fd7c.abi3.so → torch213-cxx11-cu130-x86_64-linux/_sageattention2_blackwell_cuda_5542189.abi3.so} +2 -2
- build/torch213-cxx11-cu130-x86_64-linux/metadata.json +6 -6
- build/torch213-cxx11-cu130-x86_64-linux/sageattention2_blackwell/__init__.py +0 -26
- build/torch213-cxx11-cu132-x86_64-linux/_ops.py +3 -3
- build/{torch212-cxx11-cu132-x86_64-linux/_sageattention2_blackwell_cuda_5a0fd7c.abi3.so → torch213-cxx11-cu132-x86_64-linux/_sageattention2_blackwell_cuda_5542189.abi3.so} +2 -2
- build/torch213-cxx11-cu132-x86_64-linux/metadata.json +6 -6
- build/torch213-cxx11-cu132-x86_64-linux/sageattention2_blackwell/__init__.py +0 -26
README.md
DELETED
|
@@ -1,9 +0,0 @@
|
|
| 1 |
-
# flashrt/sageattention2-blackwell
|
| 2 |
-
|
| 3 |
-
This repository is a compatibility mirror for older `kernels` clients
|
| 4 |
-
that resolve repositories through the default Hugging Face model repo API.
|
| 5 |
-
|
| 6 |
-
Canonical Kernel Hub repo: https://huggingface.co/kernels/flashrt/sageattention2-blackwell
|
| 7 |
-
|
| 8 |
-
Do not edit this mirror by hand. It is generated from the Kernel Hub
|
| 9 |
-
`vN` branches and contains the same `build/**` artifacts.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
build/torch212-cxx11-cu130-x86_64-linux/_ops.py
CHANGED
|
@@ -1,9 +1,9 @@
|
|
| 1 |
import torch
|
| 2 |
-
from . import
|
| 3 |
-
ops = torch.ops.
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
-
return f"
|
|
|
|
| 1 |
import torch
|
| 2 |
+
from . import _sageattention2_blackwell_cuda_5542189
|
| 3 |
+
ops = torch.ops._sageattention2_blackwell_cuda_5542189
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
+
return f"_sageattention2_blackwell_cuda_5542189::{op_name}"
|
build/{torch213-cxx11-cu130-x86_64-linux/_sageattention2_blackwell_cuda_5a0fd7c.abi3.so → torch212-cxx11-cu130-x86_64-linux/_sageattention2_blackwell_cuda_5542189.abi3.so}
RENAMED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:890ba58ff538a10e9f653567cc73980549c19e2359ec294f12b626485af1ed70
|
| 3 |
+
size 1942872
|
build/torch212-cxx11-cu130-x86_64-linux/metadata.json
CHANGED
|
@@ -1,9 +1,10 @@
|
|
| 1 |
{
|
| 2 |
"name": "sageattention2-blackwell",
|
| 3 |
-
"id": "
|
| 4 |
"version": 1,
|
| 5 |
"license": "Apache-2.0",
|
| 6 |
"python-depends": [],
|
|
|
|
| 7 |
"backend": {
|
| 8 |
"type": "cuda",
|
| 9 |
"archs": [
|
|
@@ -15,19 +16,18 @@
|
|
| 15 |
"algorithm": "sha256",
|
| 16 |
"files": {
|
| 17 |
"__init__.py": "8RunKSRFou+DKdhegD8QAZyIVl14pLifEwNYbIa72jY=",
|
| 18 |
-
"_ops.py": "
|
| 19 |
-
"
|
| 20 |
-
"sageattention2_blackwell/__init__.py": "DFYPlrhXwYjEqCl/8n0SmWGZV8NFml5DPhMjKfv98GY="
|
| 21 |
}
|
| 22 |
},
|
| 23 |
"provenance": {
|
| 24 |
"kernel-builder": {
|
| 25 |
"version": "0.17.0-dev0",
|
| 26 |
-
"sha": "
|
| 27 |
"dirty": false
|
| 28 |
},
|
| 29 |
"kernel": {
|
| 30 |
-
"sha": "
|
| 31 |
"dirty": false
|
| 32 |
}
|
| 33 |
}
|
|
|
|
| 1 |
{
|
| 2 |
"name": "sageattention2-blackwell",
|
| 3 |
+
"id": "_sageattention2_blackwell_cuda_5542189",
|
| 4 |
"version": 1,
|
| 5 |
"license": "Apache-2.0",
|
| 6 |
"python-depends": [],
|
| 7 |
+
"kernel-depends": [],
|
| 8 |
"backend": {
|
| 9 |
"type": "cuda",
|
| 10 |
"archs": [
|
|
|
|
| 16 |
"algorithm": "sha256",
|
| 17 |
"files": {
|
| 18 |
"__init__.py": "8RunKSRFou+DKdhegD8QAZyIVl14pLifEwNYbIa72jY=",
|
| 19 |
+
"_ops.py": "pDaNx2t5PnKVEVqTSd3ensPGiyrHpyDIzxhNCbNH9FY=",
|
| 20 |
+
"_sageattention2_blackwell_cuda_5542189.abi3.so": "iQulj/U4oQ6fZTVnzHOYBUnBniNZ7ClPErYmSFrx7XA="
|
|
|
|
| 21 |
}
|
| 22 |
},
|
| 23 |
"provenance": {
|
| 24 |
"kernel-builder": {
|
| 25 |
"version": "0.17.0-dev0",
|
| 26 |
+
"sha": "3a5c09d941a19f582fe551d038dc2b3bc6b0a03d",
|
| 27 |
"dirty": false
|
| 28 |
},
|
| 29 |
"kernel": {
|
| 30 |
+
"sha": "5542189ca3e6593cabfd117a3b1becf0661c60d3",
|
| 31 |
"dirty": false
|
| 32 |
}
|
| 33 |
}
|
build/torch212-cxx11-cu130-x86_64-linux/sageattention2_blackwell/__init__.py
DELETED
|
@@ -1,26 +0,0 @@
|
|
| 1 |
-
import ctypes
|
| 2 |
-
import importlib.util
|
| 3 |
-
import sys
|
| 4 |
-
from pathlib import Path
|
| 5 |
-
from types import ModuleType
|
| 6 |
-
|
| 7 |
-
|
| 8 |
-
def _import_from_path(file_path: Path) -> ModuleType:
|
| 9 |
-
# We cannot use the module name as-is, after adding it to `sys.modules`,
|
| 10 |
-
# it would also be used for other imports. So, we make a module name that
|
| 11 |
-
# depends on the path for it to be unique using the hex-encoded hash of
|
| 12 |
-
# the path.
|
| 13 |
-
path_hash = "{:x}".format(ctypes.c_size_t(hash(file_path.absolute())).value)
|
| 14 |
-
module_name = path_hash
|
| 15 |
-
spec = importlib.util.spec_from_file_location(module_name, file_path)
|
| 16 |
-
if spec is None:
|
| 17 |
-
raise ImportError(f"Cannot load spec for {module_name} from {file_path}")
|
| 18 |
-
module = importlib.util.module_from_spec(spec)
|
| 19 |
-
if module is None:
|
| 20 |
-
raise ImportError(f"Cannot load module {module_name} from spec")
|
| 21 |
-
sys.modules[module_name] = module
|
| 22 |
-
spec.loader.exec_module(module) # type: ignore
|
| 23 |
-
return module
|
| 24 |
-
|
| 25 |
-
|
| 26 |
-
globals().update(vars(_import_from_path(Path(__file__).parent.parent / "__init__.py")))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
build/torch212-cxx11-cu132-x86_64-linux/_ops.py
CHANGED
|
@@ -1,9 +1,9 @@
|
|
| 1 |
import torch
|
| 2 |
-
from . import
|
| 3 |
-
ops = torch.ops.
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
-
return f"
|
|
|
|
| 1 |
import torch
|
| 2 |
+
from . import _sageattention2_blackwell_cuda_5542189
|
| 3 |
+
ops = torch.ops._sageattention2_blackwell_cuda_5542189
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
+
return f"_sageattention2_blackwell_cuda_5542189::{op_name}"
|
build/{torch212-cxx11-cu130-x86_64-linux/_sageattention2_blackwell_cuda_5a0fd7c.abi3.so → torch212-cxx11-cu132-x86_64-linux/_sageattention2_blackwell_cuda_5542189.abi3.so}
RENAMED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3e0bf409f82574c40d33c63b20950f5b898f6ecaad8a8f8e2cf7db638afd804e
|
| 3 |
+
size 1799616
|
build/torch212-cxx11-cu132-x86_64-linux/metadata.json
CHANGED
|
@@ -1,9 +1,10 @@
|
|
| 1 |
{
|
| 2 |
"name": "sageattention2-blackwell",
|
| 3 |
-
"id": "
|
| 4 |
"version": 1,
|
| 5 |
"license": "Apache-2.0",
|
| 6 |
"python-depends": [],
|
|
|
|
| 7 |
"backend": {
|
| 8 |
"type": "cuda",
|
| 9 |
"archs": [
|
|
@@ -15,19 +16,18 @@
|
|
| 15 |
"algorithm": "sha256",
|
| 16 |
"files": {
|
| 17 |
"__init__.py": "8RunKSRFou+DKdhegD8QAZyIVl14pLifEwNYbIa72jY=",
|
| 18 |
-
"_ops.py": "
|
| 19 |
-
"
|
| 20 |
-
"sageattention2_blackwell/__init__.py": "DFYPlrhXwYjEqCl/8n0SmWGZV8NFml5DPhMjKfv98GY="
|
| 21 |
}
|
| 22 |
},
|
| 23 |
"provenance": {
|
| 24 |
"kernel-builder": {
|
| 25 |
"version": "0.17.0-dev0",
|
| 26 |
-
"sha": "
|
| 27 |
"dirty": false
|
| 28 |
},
|
| 29 |
"kernel": {
|
| 30 |
-
"sha": "
|
| 31 |
"dirty": false
|
| 32 |
}
|
| 33 |
}
|
|
|
|
| 1 |
{
|
| 2 |
"name": "sageattention2-blackwell",
|
| 3 |
+
"id": "_sageattention2_blackwell_cuda_5542189",
|
| 4 |
"version": 1,
|
| 5 |
"license": "Apache-2.0",
|
| 6 |
"python-depends": [],
|
| 7 |
+
"kernel-depends": [],
|
| 8 |
"backend": {
|
| 9 |
"type": "cuda",
|
| 10 |
"archs": [
|
|
|
|
| 16 |
"algorithm": "sha256",
|
| 17 |
"files": {
|
| 18 |
"__init__.py": "8RunKSRFou+DKdhegD8QAZyIVl14pLifEwNYbIa72jY=",
|
| 19 |
+
"_ops.py": "pDaNx2t5PnKVEVqTSd3ensPGiyrHpyDIzxhNCbNH9FY=",
|
| 20 |
+
"_sageattention2_blackwell_cuda_5542189.abi3.so": "Pgv0CfgldMQNM8Y7IJUPW4mPbsqtio+OLPfbY4r9gE4="
|
|
|
|
| 21 |
}
|
| 22 |
},
|
| 23 |
"provenance": {
|
| 24 |
"kernel-builder": {
|
| 25 |
"version": "0.17.0-dev0",
|
| 26 |
+
"sha": "3a5c09d941a19f582fe551d038dc2b3bc6b0a03d",
|
| 27 |
"dirty": false
|
| 28 |
},
|
| 29 |
"kernel": {
|
| 30 |
+
"sha": "5542189ca3e6593cabfd117a3b1becf0661c60d3",
|
| 31 |
"dirty": false
|
| 32 |
}
|
| 33 |
}
|
build/torch212-cxx11-cu132-x86_64-linux/sageattention2_blackwell/__init__.py
DELETED
|
@@ -1,26 +0,0 @@
|
|
| 1 |
-
import ctypes
|
| 2 |
-
import importlib.util
|
| 3 |
-
import sys
|
| 4 |
-
from pathlib import Path
|
| 5 |
-
from types import ModuleType
|
| 6 |
-
|
| 7 |
-
|
| 8 |
-
def _import_from_path(file_path: Path) -> ModuleType:
|
| 9 |
-
# We cannot use the module name as-is, after adding it to `sys.modules`,
|
| 10 |
-
# it would also be used for other imports. So, we make a module name that
|
| 11 |
-
# depends on the path for it to be unique using the hex-encoded hash of
|
| 12 |
-
# the path.
|
| 13 |
-
path_hash = "{:x}".format(ctypes.c_size_t(hash(file_path.absolute())).value)
|
| 14 |
-
module_name = path_hash
|
| 15 |
-
spec = importlib.util.spec_from_file_location(module_name, file_path)
|
| 16 |
-
if spec is None:
|
| 17 |
-
raise ImportError(f"Cannot load spec for {module_name} from {file_path}")
|
| 18 |
-
module = importlib.util.module_from_spec(spec)
|
| 19 |
-
if module is None:
|
| 20 |
-
raise ImportError(f"Cannot load module {module_name} from spec")
|
| 21 |
-
sys.modules[module_name] = module
|
| 22 |
-
spec.loader.exec_module(module) # type: ignore
|
| 23 |
-
return module
|
| 24 |
-
|
| 25 |
-
|
| 26 |
-
globals().update(vars(_import_from_path(Path(__file__).parent.parent / "__init__.py")))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
build/torch213-cxx11-cu130-x86_64-linux/_ops.py
CHANGED
|
@@ -1,9 +1,9 @@
|
|
| 1 |
import torch
|
| 2 |
-
from . import
|
| 3 |
-
ops = torch.ops.
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
-
return f"
|
|
|
|
| 1 |
import torch
|
| 2 |
+
from . import _sageattention2_blackwell_cuda_5542189
|
| 3 |
+
ops = torch.ops._sageattention2_blackwell_cuda_5542189
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
+
return f"_sageattention2_blackwell_cuda_5542189::{op_name}"
|
build/{torch213-cxx11-cu132-x86_64-linux/_sageattention2_blackwell_cuda_5a0fd7c.abi3.so → torch213-cxx11-cu130-x86_64-linux/_sageattention2_blackwell_cuda_5542189.abi3.so}
RENAMED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:438f5764f533bee2f7c970a837789a9722d5c8841399d4fa8f5e4eee145508fb
|
| 3 |
+
size 1942712
|
build/torch213-cxx11-cu130-x86_64-linux/metadata.json
CHANGED
|
@@ -1,9 +1,10 @@
|
|
| 1 |
{
|
| 2 |
"name": "sageattention2-blackwell",
|
| 3 |
-
"id": "
|
| 4 |
"version": 1,
|
| 5 |
"license": "Apache-2.0",
|
| 6 |
"python-depends": [],
|
|
|
|
| 7 |
"backend": {
|
| 8 |
"type": "cuda",
|
| 9 |
"archs": [
|
|
@@ -15,19 +16,18 @@
|
|
| 15 |
"algorithm": "sha256",
|
| 16 |
"files": {
|
| 17 |
"__init__.py": "8RunKSRFou+DKdhegD8QAZyIVl14pLifEwNYbIa72jY=",
|
| 18 |
-
"_ops.py": "
|
| 19 |
-
"
|
| 20 |
-
"sageattention2_blackwell/__init__.py": "DFYPlrhXwYjEqCl/8n0SmWGZV8NFml5DPhMjKfv98GY="
|
| 21 |
}
|
| 22 |
},
|
| 23 |
"provenance": {
|
| 24 |
"kernel-builder": {
|
| 25 |
"version": "0.17.0-dev0",
|
| 26 |
-
"sha": "
|
| 27 |
"dirty": false
|
| 28 |
},
|
| 29 |
"kernel": {
|
| 30 |
-
"sha": "
|
| 31 |
"dirty": false
|
| 32 |
}
|
| 33 |
}
|
|
|
|
| 1 |
{
|
| 2 |
"name": "sageattention2-blackwell",
|
| 3 |
+
"id": "_sageattention2_blackwell_cuda_5542189",
|
| 4 |
"version": 1,
|
| 5 |
"license": "Apache-2.0",
|
| 6 |
"python-depends": [],
|
| 7 |
+
"kernel-depends": [],
|
| 8 |
"backend": {
|
| 9 |
"type": "cuda",
|
| 10 |
"archs": [
|
|
|
|
| 16 |
"algorithm": "sha256",
|
| 17 |
"files": {
|
| 18 |
"__init__.py": "8RunKSRFou+DKdhegD8QAZyIVl14pLifEwNYbIa72jY=",
|
| 19 |
+
"_ops.py": "pDaNx2t5PnKVEVqTSd3ensPGiyrHpyDIzxhNCbNH9FY=",
|
| 20 |
+
"_sageattention2_blackwell_cuda_5542189.abi3.so": "Q49XZPUzvuL3yXCoN3ialyLVyIQTmdT6j15O7hRVCPs="
|
|
|
|
| 21 |
}
|
| 22 |
},
|
| 23 |
"provenance": {
|
| 24 |
"kernel-builder": {
|
| 25 |
"version": "0.17.0-dev0",
|
| 26 |
+
"sha": "3a5c09d941a19f582fe551d038dc2b3bc6b0a03d",
|
| 27 |
"dirty": false
|
| 28 |
},
|
| 29 |
"kernel": {
|
| 30 |
+
"sha": "5542189ca3e6593cabfd117a3b1becf0661c60d3",
|
| 31 |
"dirty": false
|
| 32 |
}
|
| 33 |
}
|
build/torch213-cxx11-cu130-x86_64-linux/sageattention2_blackwell/__init__.py
DELETED
|
@@ -1,26 +0,0 @@
|
|
| 1 |
-
import ctypes
|
| 2 |
-
import importlib.util
|
| 3 |
-
import sys
|
| 4 |
-
from pathlib import Path
|
| 5 |
-
from types import ModuleType
|
| 6 |
-
|
| 7 |
-
|
| 8 |
-
def _import_from_path(file_path: Path) -> ModuleType:
|
| 9 |
-
# We cannot use the module name as-is, after adding it to `sys.modules`,
|
| 10 |
-
# it would also be used for other imports. So, we make a module name that
|
| 11 |
-
# depends on the path for it to be unique using the hex-encoded hash of
|
| 12 |
-
# the path.
|
| 13 |
-
path_hash = "{:x}".format(ctypes.c_size_t(hash(file_path.absolute())).value)
|
| 14 |
-
module_name = path_hash
|
| 15 |
-
spec = importlib.util.spec_from_file_location(module_name, file_path)
|
| 16 |
-
if spec is None:
|
| 17 |
-
raise ImportError(f"Cannot load spec for {module_name} from {file_path}")
|
| 18 |
-
module = importlib.util.module_from_spec(spec)
|
| 19 |
-
if module is None:
|
| 20 |
-
raise ImportError(f"Cannot load module {module_name} from spec")
|
| 21 |
-
sys.modules[module_name] = module
|
| 22 |
-
spec.loader.exec_module(module) # type: ignore
|
| 23 |
-
return module
|
| 24 |
-
|
| 25 |
-
|
| 26 |
-
globals().update(vars(_import_from_path(Path(__file__).parent.parent / "__init__.py")))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
build/torch213-cxx11-cu132-x86_64-linux/_ops.py
CHANGED
|
@@ -1,9 +1,9 @@
|
|
| 1 |
import torch
|
| 2 |
-
from . import
|
| 3 |
-
ops = torch.ops.
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
-
return f"
|
|
|
|
| 1 |
import torch
|
| 2 |
+
from . import _sageattention2_blackwell_cuda_5542189
|
| 3 |
+
ops = torch.ops._sageattention2_blackwell_cuda_5542189
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
+
return f"_sageattention2_blackwell_cuda_5542189::{op_name}"
|
build/{torch212-cxx11-cu132-x86_64-linux/_sageattention2_blackwell_cuda_5a0fd7c.abi3.so → torch213-cxx11-cu132-x86_64-linux/_sageattention2_blackwell_cuda_5542189.abi3.so}
RENAMED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3a57447f1952300a6b372a8711cc8ac75640ed108a18bf8955f6744b8bd0c926
|
| 3 |
+
size 1799456
|
build/torch213-cxx11-cu132-x86_64-linux/metadata.json
CHANGED
|
@@ -1,9 +1,10 @@
|
|
| 1 |
{
|
| 2 |
"name": "sageattention2-blackwell",
|
| 3 |
-
"id": "
|
| 4 |
"version": 1,
|
| 5 |
"license": "Apache-2.0",
|
| 6 |
"python-depends": [],
|
|
|
|
| 7 |
"backend": {
|
| 8 |
"type": "cuda",
|
| 9 |
"archs": [
|
|
@@ -15,19 +16,18 @@
|
|
| 15 |
"algorithm": "sha256",
|
| 16 |
"files": {
|
| 17 |
"__init__.py": "8RunKSRFou+DKdhegD8QAZyIVl14pLifEwNYbIa72jY=",
|
| 18 |
-
"_ops.py": "
|
| 19 |
-
"
|
| 20 |
-
"sageattention2_blackwell/__init__.py": "DFYPlrhXwYjEqCl/8n0SmWGZV8NFml5DPhMjKfv98GY="
|
| 21 |
}
|
| 22 |
},
|
| 23 |
"provenance": {
|
| 24 |
"kernel-builder": {
|
| 25 |
"version": "0.17.0-dev0",
|
| 26 |
-
"sha": "
|
| 27 |
"dirty": false
|
| 28 |
},
|
| 29 |
"kernel": {
|
| 30 |
-
"sha": "
|
| 31 |
"dirty": false
|
| 32 |
}
|
| 33 |
}
|
|
|
|
| 1 |
{
|
| 2 |
"name": "sageattention2-blackwell",
|
| 3 |
+
"id": "_sageattention2_blackwell_cuda_5542189",
|
| 4 |
"version": 1,
|
| 5 |
"license": "Apache-2.0",
|
| 6 |
"python-depends": [],
|
| 7 |
+
"kernel-depends": [],
|
| 8 |
"backend": {
|
| 9 |
"type": "cuda",
|
| 10 |
"archs": [
|
|
|
|
| 16 |
"algorithm": "sha256",
|
| 17 |
"files": {
|
| 18 |
"__init__.py": "8RunKSRFou+DKdhegD8QAZyIVl14pLifEwNYbIa72jY=",
|
| 19 |
+
"_ops.py": "pDaNx2t5PnKVEVqTSd3ensPGiyrHpyDIzxhNCbNH9FY=",
|
| 20 |
+
"_sageattention2_blackwell_cuda_5542189.abi3.so": "OldEfxlSMAprNyqHEcyKx1ZA7RCKGL+JVfZ0S4vQySY="
|
|
|
|
| 21 |
}
|
| 22 |
},
|
| 23 |
"provenance": {
|
| 24 |
"kernel-builder": {
|
| 25 |
"version": "0.17.0-dev0",
|
| 26 |
+
"sha": "3a5c09d941a19f582fe551d038dc2b3bc6b0a03d",
|
| 27 |
"dirty": false
|
| 28 |
},
|
| 29 |
"kernel": {
|
| 30 |
+
"sha": "5542189ca3e6593cabfd117a3b1becf0661c60d3",
|
| 31 |
"dirty": false
|
| 32 |
}
|
| 33 |
}
|
build/torch213-cxx11-cu132-x86_64-linux/sageattention2_blackwell/__init__.py
DELETED
|
@@ -1,26 +0,0 @@
|
|
| 1 |
-
import ctypes
|
| 2 |
-
import importlib.util
|
| 3 |
-
import sys
|
| 4 |
-
from pathlib import Path
|
| 5 |
-
from types import ModuleType
|
| 6 |
-
|
| 7 |
-
|
| 8 |
-
def _import_from_path(file_path: Path) -> ModuleType:
|
| 9 |
-
# We cannot use the module name as-is, after adding it to `sys.modules`,
|
| 10 |
-
# it would also be used for other imports. So, we make a module name that
|
| 11 |
-
# depends on the path for it to be unique using the hex-encoded hash of
|
| 12 |
-
# the path.
|
| 13 |
-
path_hash = "{:x}".format(ctypes.c_size_t(hash(file_path.absolute())).value)
|
| 14 |
-
module_name = path_hash
|
| 15 |
-
spec = importlib.util.spec_from_file_location(module_name, file_path)
|
| 16 |
-
if spec is None:
|
| 17 |
-
raise ImportError(f"Cannot load spec for {module_name} from {file_path}")
|
| 18 |
-
module = importlib.util.module_from_spec(spec)
|
| 19 |
-
if module is None:
|
| 20 |
-
raise ImportError(f"Cannot load module {module_name} from spec")
|
| 21 |
-
sys.modules[module_name] = module
|
| 22 |
-
spec.loader.exec_module(module) # type: ignore
|
| 23 |
-
return module
|
| 24 |
-
|
| 25 |
-
|
| 26 |
-
globals().update(vars(_import_from_path(Path(__file__).parent.parent / "__init__.py")))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|