Hugging Face's logo

Kernels:
kernels-community
/
mra
Trusted publisher

Kernel card Files
xet
 Community
kernels-bot commited on
Commit
f9bb240
·
verified ·
1 Parent(s): 30edbcf

Uploaded using `kernel-builder`.

Browse files
Files changed (34) hide show
  1. .gitattributes +9 -0
  2. build/torch210-cxx11-cu126-aarch64-linux/{_mra_cuda_86f75d9.abi3.so → _mra_cuda_10efe7e.abi3.so} +1 -1
  3. build/torch210-cxx11-cu126-aarch64-linux/_ops.py +3 -3
  4. build/torch210-cxx11-cu126-aarch64-linux/metadata.json +1 -1
  5. build/{torch211-cxx11-cu128-aarch64-linux/_mra_cuda_86f75d9.abi3.so → torch210-cxx11-cu128-aarch64-linux/_mra_cuda_10efe7e.abi3.so} +1 -1
  6. build/torch210-cxx11-cu128-aarch64-linux/_ops.py +3 -3
  7. build/torch210-cxx11-cu128-aarch64-linux/metadata.json +1 -1
  8. build/{torch211-cxx11-cu130-aarch64-linux/_mra_cuda_86f75d9.abi3.so → torch210-cxx11-cu130-aarch64-linux/_mra_cuda_10efe7e.abi3.so} +1 -1
  9. build/torch210-cxx11-cu130-aarch64-linux/_ops.py +3 -3
  10. build/torch210-cxx11-cu130-aarch64-linux/metadata.json +1 -1
  11. build/torch211-cxx11-cu126-aarch64-linux/{_mra_cuda_86f75d9.abi3.so → _mra_cuda_10efe7e.abi3.so} +1 -1
  12. build/torch211-cxx11-cu126-aarch64-linux/_ops.py +3 -3
  13. build/torch211-cxx11-cu126-aarch64-linux/metadata.json +1 -1
  14. build/{torch210-cxx11-cu128-aarch64-linux/_mra_cuda_86f75d9.abi3.so → torch211-cxx11-cu128-aarch64-linux/_mra_cuda_10efe7e.abi3.so} +1 -1
  15. build/torch211-cxx11-cu128-aarch64-linux/_ops.py +3 -3
  16. build/torch211-cxx11-cu128-aarch64-linux/metadata.json +1 -1
  17. build/{torch210-cxx11-cu130-aarch64-linux/_mra_cuda_86f75d9.abi3.so → torch211-cxx11-cu130-aarch64-linux/_mra_cuda_10efe7e.abi3.so} +1 -1
  18. build/torch211-cxx11-cu130-aarch64-linux/_ops.py +3 -3
  19. build/torch211-cxx11-cu130-aarch64-linux/metadata.json +1 -1
  20. build/torch212-cxx11-cu126-aarch64-linux/__init__.py +25 -0
  21. build/torch212-cxx11-cu126-aarch64-linux/_mra_cuda_10efe7e.abi3.so +3 -0
  22. build/torch212-cxx11-cu126-aarch64-linux/_ops.py +9 -0
  23. build/torch212-cxx11-cu126-aarch64-linux/metadata.json +20 -0
  24. build/torch212-cxx11-cu126-aarch64-linux/mra/__init__.py +26 -0
  25. build/torch212-cxx11-cu130-aarch64-linux/__init__.py +25 -0
  26. build/torch212-cxx11-cu130-aarch64-linux/_mra_cuda_10efe7e.abi3.so +3 -0
  27. build/torch212-cxx11-cu130-aarch64-linux/_ops.py +9 -0
  28. build/torch212-cxx11-cu130-aarch64-linux/metadata.json +21 -0
  29. build/torch212-cxx11-cu130-aarch64-linux/mra/__init__.py +26 -0
  30. build/torch212-cxx11-cu132-aarch64-linux/__init__.py +25 -0
  31. build/torch212-cxx11-cu132-aarch64-linux/_mra_cuda_10efe7e.abi3.so +3 -0
  32. build/torch212-cxx11-cu132-aarch64-linux/_ops.py +9 -0
  33. build/torch212-cxx11-cu132-aarch64-linux/metadata.json +21 -0
  34. build/torch212-cxx11-cu132-aarch64-linux/mra/__init__.py +26 -0
.gitattributes CHANGED
@@ -156,3 +156,12 @@ build/torch210-cxx11-cu130-x86_64-linux/_mra_cuda_86f75d9.abi3.so filter=lfs dif
156
  build/torch211-cxx11-cu126-x86_64-linux/_mra_cuda_86f75d9.abi3.so filter=lfs diff=lfs merge=lfs -text
157
  build/torch211-cxx11-cu128-x86_64-linux/_mra_cuda_86f75d9.abi3.so filter=lfs diff=lfs merge=lfs -text
158
  build/torch211-cxx11-cu130-x86_64-linux/_mra_cuda_86f75d9.abi3.so filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
 
 
 
 
 
156
  build/torch211-cxx11-cu126-x86_64-linux/_mra_cuda_86f75d9.abi3.so filter=lfs diff=lfs merge=lfs -text
157
  build/torch211-cxx11-cu128-x86_64-linux/_mra_cuda_86f75d9.abi3.so filter=lfs diff=lfs merge=lfs -text
158
  build/torch211-cxx11-cu130-x86_64-linux/_mra_cuda_86f75d9.abi3.so filter=lfs diff=lfs merge=lfs -text
159
+ build/torch210-cxx11-cu126-aarch64-linux/_mra_cuda_10efe7e.abi3.so filter=lfs diff=lfs merge=lfs -text
160
+ build/torch210-cxx11-cu128-aarch64-linux/_mra_cuda_10efe7e.abi3.so filter=lfs diff=lfs merge=lfs -text
161
+ build/torch210-cxx11-cu130-aarch64-linux/_mra_cuda_10efe7e.abi3.so filter=lfs diff=lfs merge=lfs -text
162
+ build/torch211-cxx11-cu126-aarch64-linux/_mra_cuda_10efe7e.abi3.so filter=lfs diff=lfs merge=lfs -text
163
+ build/torch211-cxx11-cu128-aarch64-linux/_mra_cuda_10efe7e.abi3.so filter=lfs diff=lfs merge=lfs -text
164
+ build/torch211-cxx11-cu130-aarch64-linux/_mra_cuda_10efe7e.abi3.so filter=lfs diff=lfs merge=lfs -text
165
+ build/torch212-cxx11-cu126-aarch64-linux/_mra_cuda_10efe7e.abi3.so filter=lfs diff=lfs merge=lfs -text
166
+ build/torch212-cxx11-cu130-aarch64-linux/_mra_cuda_10efe7e.abi3.so filter=lfs diff=lfs merge=lfs -text
167
+ build/torch212-cxx11-cu132-aarch64-linux/_mra_cuda_10efe7e.abi3.so filter=lfs diff=lfs merge=lfs -text
build/torch210-cxx11-cu126-aarch64-linux/{_mra_cuda_86f75d9.abi3.so → _mra_cuda_10efe7e.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a42e36348e8e1834f95384d0b9e66821bbc86af06aed76233fa0a8f45a199dc3
3
  size 2567952
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fcd3778eed080eb40aa01797511d09881b7193fb3f696e052540330c9a4285a9
3
  size 2567952
build/torch210-cxx11-cu126-aarch64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _mra_cuda_86f75d9
3
- ops = torch.ops._mra_cuda_86f75d9
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_mra_cuda_86f75d9::{op_name}"
 
1
  import torch
2
+ from . import _mra_cuda_10efe7e
3
+ ops = torch.ops._mra_cuda_10efe7e
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_mra_cuda_10efe7e::{op_name}"
build/torch210-cxx11-cu126-aarch64-linux/metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "name": "mra",
3
- "id": "_mra_cuda_86f75d9",
4
  "version": 1,
5
  "license": "Apache-2.0",
6
  "python-depends": [],
 
1
  {
2
  "name": "mra",
3
+ "id": "_mra_cuda_10efe7e",
4
  "version": 1,
5
  "license": "Apache-2.0",
6
  "python-depends": [],
build/{torch211-cxx11-cu128-aarch64-linux/_mra_cuda_86f75d9.abi3.so → torch210-cxx11-cu128-aarch64-linux/_mra_cuda_10efe7e.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2fe42dd5dde0f637f665a9ef673828c7dd09526bf6a019673347448656591217
3
  size 2830296
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ec34652c3a8791be9866b0fc3056ed24c389dc2131a22cdde13fbdea118dabcb
3
  size 2830296
build/torch210-cxx11-cu128-aarch64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _mra_cuda_86f75d9
3
- ops = torch.ops._mra_cuda_86f75d9
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_mra_cuda_86f75d9::{op_name}"
 
1
  import torch
2
+ from . import _mra_cuda_10efe7e
3
+ ops = torch.ops._mra_cuda_10efe7e
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_mra_cuda_10efe7e::{op_name}"
build/torch210-cxx11-cu128-aarch64-linux/metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "name": "mra",
3
- "id": "_mra_cuda_86f75d9",
4
  "version": 1,
5
  "license": "Apache-2.0",
6
  "python-depends": [],
 
1
  {
2
  "name": "mra",
3
+ "id": "_mra_cuda_10efe7e",
4
  "version": 1,
5
  "license": "Apache-2.0",
6
  "python-depends": [],
build/{torch211-cxx11-cu130-aarch64-linux/_mra_cuda_86f75d9.abi3.so → torch210-cxx11-cu130-aarch64-linux/_mra_cuda_10efe7e.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:fdf03fb23908b1168e46d13013195f15e43cd214b85f26ec6b2c4f95f735d81c
3
  size 2768280
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2fdf113e060cbb4cd22c5c1077752122282317b41bc34692d4198243daf43c18
3
  size 2768280
build/torch210-cxx11-cu130-aarch64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _mra_cuda_86f75d9
3
- ops = torch.ops._mra_cuda_86f75d9
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_mra_cuda_86f75d9::{op_name}"
 
1
  import torch
2
+ from . import _mra_cuda_10efe7e
3
+ ops = torch.ops._mra_cuda_10efe7e
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_mra_cuda_10efe7e::{op_name}"
build/torch210-cxx11-cu130-aarch64-linux/metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "name": "mra",
3
- "id": "_mra_cuda_86f75d9",
4
  "version": 1,
5
  "license": "Apache-2.0",
6
  "python-depends": [],
 
1
  {
2
  "name": "mra",
3
+ "id": "_mra_cuda_10efe7e",
4
  "version": 1,
5
  "license": "Apache-2.0",
6
  "python-depends": [],
build/torch211-cxx11-cu126-aarch64-linux/{_mra_cuda_86f75d9.abi3.so → _mra_cuda_10efe7e.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e8dd4bebf8f4d196371a17ce8ffc763c60a881c959752dd3df931d99252b8cc0
3
  size 2567952
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eaeb0ea45eb82dca35a5193e952054188660378e326ebd4f535bbb5dc35051bd
3
  size 2567952
build/torch211-cxx11-cu126-aarch64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _mra_cuda_86f75d9
3
- ops = torch.ops._mra_cuda_86f75d9
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_mra_cuda_86f75d9::{op_name}"
 
1
  import torch
2
+ from . import _mra_cuda_10efe7e
3
+ ops = torch.ops._mra_cuda_10efe7e
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_mra_cuda_10efe7e::{op_name}"
build/torch211-cxx11-cu126-aarch64-linux/metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "name": "mra",
3
- "id": "_mra_cuda_86f75d9",
4
  "version": 1,
5
  "license": "Apache-2.0",
6
  "python-depends": [],
 
1
  {
2
  "name": "mra",
3
+ "id": "_mra_cuda_10efe7e",
4
  "version": 1,
5
  "license": "Apache-2.0",
6
  "python-depends": [],
build/{torch210-cxx11-cu128-aarch64-linux/_mra_cuda_86f75d9.abi3.so → torch211-cxx11-cu128-aarch64-linux/_mra_cuda_10efe7e.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6deceb98946253acc87be3fd22a19705b3dab1868282e47f4006197c92af420d
3
  size 2830296
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:905af830a66620a25573dd290aee003ccf18152079ecd04762431b9395353bb2
3
  size 2830296
build/torch211-cxx11-cu128-aarch64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _mra_cuda_86f75d9
3
- ops = torch.ops._mra_cuda_86f75d9
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_mra_cuda_86f75d9::{op_name}"
 
1
  import torch
2
+ from . import _mra_cuda_10efe7e
3
+ ops = torch.ops._mra_cuda_10efe7e
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_mra_cuda_10efe7e::{op_name}"
build/torch211-cxx11-cu128-aarch64-linux/metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "name": "mra",
3
- "id": "_mra_cuda_86f75d9",
4
  "version": 1,
5
  "license": "Apache-2.0",
6
  "python-depends": [],
 
1
  {
2
  "name": "mra",
3
+ "id": "_mra_cuda_10efe7e",
4
  "version": 1,
5
  "license": "Apache-2.0",
6
  "python-depends": [],
build/{torch210-cxx11-cu130-aarch64-linux/_mra_cuda_86f75d9.abi3.so → torch211-cxx11-cu130-aarch64-linux/_mra_cuda_10efe7e.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f0b0077374c962a94d015014a9dac579002a49ca17fc72d21efa86c7804a4351
3
  size 2768280
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:42f101d864e1d408907ce349fa6df024be48592e11e6dcb5c9c5f279d9275e2b
3
  size 2768280
build/torch211-cxx11-cu130-aarch64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _mra_cuda_86f75d9
3
- ops = torch.ops._mra_cuda_86f75d9
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_mra_cuda_86f75d9::{op_name}"
 
1
  import torch
2
+ from . import _mra_cuda_10efe7e
3
+ ops = torch.ops._mra_cuda_10efe7e
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_mra_cuda_10efe7e::{op_name}"
build/torch211-cxx11-cu130-aarch64-linux/metadata.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "name": "mra",
3
- "id": "_mra_cuda_86f75d9",
4
  "version": 1,
5
  "license": "Apache-2.0",
6
  "python-depends": [],
 
1
  {
2
  "name": "mra",
3
+ "id": "_mra_cuda_10efe7e",
4
  "version": 1,
5
  "license": "Apache-2.0",
6
  "python-depends": [],
build/torch212-cxx11-cu126-aarch64-linux/__init__.py ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from ._ops import ops
2
+ import torch
3
+
4
+ def index_max(index_vals: torch.Tensor, indices: torch.Tensor, A_num_block: int, B_num_block: int):
5
+ return ops.index_max(index_vals, indices, A_num_block, B_num_block)
6
+
7
+ def mm_to_sparse(dense_A: torch.Tensor, dense_B: torch.Tensor, indices: torch.Tensor):
8
+ return ops.mm_to_sparse(dense_A, dense_B, indices)
9
+
10
+ def sparse_dense_mm(sparse_A: torch.Tensor, indices: torch.Tensor, dense_B: torch.Tensor, A_num_block: int):
11
+ return ops.sparse_dense_mm(sparse_A, indices, dense_B, A_num_block)
12
+
13
+ def reduce_sum(sparse_A: torch.Tensor, indices: torch.Tensor, A_num_block: int, B_num_block: int):
14
+ return ops.reduce_sum(sparse_A, indices, A_num_block, B_num_block)
15
+
16
+ def scatter(dense_A: torch.Tensor, indices: torch.Tensor, B_num_block: int):
17
+ return ops.scatter(dense_A, indices, B_num_block)
18
+
19
+ __all__ = [
20
+ "index_max",
21
+ "mm_to_sparse",
22
+ "sparse_dense_mm",
23
+ "reduce_sum",
24
+ "scatter",
25
+ ]
build/torch212-cxx11-cu126-aarch64-linux/_mra_cuda_10efe7e.abi3.so ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1ddcd0eb36334fe56b4458bb23c058f5f0b0df1218bacb29bc33d78fccd4aedc
3
+ size 2567992
build/torch212-cxx11-cu126-aarch64-linux/_ops.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ from . import _mra_cuda_10efe7e
3
+ ops = torch.ops._mra_cuda_10efe7e
4
+
5
+ def add_op_namespace_prefix(op_name: str):
6
+ """
7
+ Prefix op by namespace.
8
+ """
9
+ return f"_mra_cuda_10efe7e::{op_name}"
build/torch212-cxx11-cu126-aarch64-linux/metadata.json ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "mra",
3
+ "id": "_mra_cuda_10efe7e",
4
+ "version": 1,
5
+ "license": "Apache-2.0",
6
+ "python-depends": [],
7
+ "backend": {
8
+ "type": "cuda",
9
+ "archs": [
10
+ "7.0",
11
+ "7.2",
12
+ "7.5",
13
+ "8.0",
14
+ "8.6",
15
+ "8.7",
16
+ "8.9",
17
+ "9.0+PTX"
18
+ ]
19
+ }
20
+ }
build/torch212-cxx11-cu126-aarch64-linux/mra/__init__.py ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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-cu130-aarch64-linux/__init__.py ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from ._ops import ops
2
+ import torch
3
+
4
+ def index_max(index_vals: torch.Tensor, indices: torch.Tensor, A_num_block: int, B_num_block: int):
5
+ return ops.index_max(index_vals, indices, A_num_block, B_num_block)
6
+
7
+ def mm_to_sparse(dense_A: torch.Tensor, dense_B: torch.Tensor, indices: torch.Tensor):
8
+ return ops.mm_to_sparse(dense_A, dense_B, indices)
9
+
10
+ def sparse_dense_mm(sparse_A: torch.Tensor, indices: torch.Tensor, dense_B: torch.Tensor, A_num_block: int):
11
+ return ops.sparse_dense_mm(sparse_A, indices, dense_B, A_num_block)
12
+
13
+ def reduce_sum(sparse_A: torch.Tensor, indices: torch.Tensor, A_num_block: int, B_num_block: int):
14
+ return ops.reduce_sum(sparse_A, indices, A_num_block, B_num_block)
15
+
16
+ def scatter(dense_A: torch.Tensor, indices: torch.Tensor, B_num_block: int):
17
+ return ops.scatter(dense_A, indices, B_num_block)
18
+
19
+ __all__ = [
20
+ "index_max",
21
+ "mm_to_sparse",
22
+ "sparse_dense_mm",
23
+ "reduce_sum",
24
+ "scatter",
25
+ ]
build/torch212-cxx11-cu130-aarch64-linux/_mra_cuda_10efe7e.abi3.so ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2e71e37a56cc19fdd0facb3c27f5c6ddfe9c2a9e63166e4e32c2217341ee8f5b
3
+ size 2768296
build/torch212-cxx11-cu130-aarch64-linux/_ops.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ from . import _mra_cuda_10efe7e
3
+ ops = torch.ops._mra_cuda_10efe7e
4
+
5
+ def add_op_namespace_prefix(op_name: str):
6
+ """
7
+ Prefix op by namespace.
8
+ """
9
+ return f"_mra_cuda_10efe7e::{op_name}"
build/torch212-cxx11-cu130-aarch64-linux/metadata.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "mra",
3
+ "id": "_mra_cuda_10efe7e",
4
+ "version": 1,
5
+ "license": "Apache-2.0",
6
+ "python-depends": [],
7
+ "backend": {
8
+ "type": "cuda",
9
+ "archs": [
10
+ "10.0",
11
+ "11.0",
12
+ "12.0+PTX",
13
+ "7.5",
14
+ "8.0",
15
+ "8.6",
16
+ "8.7",
17
+ "8.9",
18
+ "9.0"
19
+ ]
20
+ }
21
+ }
build/torch212-cxx11-cu130-aarch64-linux/mra/__init__.py ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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-aarch64-linux/__init__.py ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from ._ops import ops
2
+ import torch
3
+
4
+ def index_max(index_vals: torch.Tensor, indices: torch.Tensor, A_num_block: int, B_num_block: int):
5
+ return ops.index_max(index_vals, indices, A_num_block, B_num_block)
6
+
7
+ def mm_to_sparse(dense_A: torch.Tensor, dense_B: torch.Tensor, indices: torch.Tensor):
8
+ return ops.mm_to_sparse(dense_A, dense_B, indices)
9
+
10
+ def sparse_dense_mm(sparse_A: torch.Tensor, indices: torch.Tensor, dense_B: torch.Tensor, A_num_block: int):
11
+ return ops.sparse_dense_mm(sparse_A, indices, dense_B, A_num_block)
12
+
13
+ def reduce_sum(sparse_A: torch.Tensor, indices: torch.Tensor, A_num_block: int, B_num_block: int):
14
+ return ops.reduce_sum(sparse_A, indices, A_num_block, B_num_block)
15
+
16
+ def scatter(dense_A: torch.Tensor, indices: torch.Tensor, B_num_block: int):
17
+ return ops.scatter(dense_A, indices, B_num_block)
18
+
19
+ __all__ = [
20
+ "index_max",
21
+ "mm_to_sparse",
22
+ "sparse_dense_mm",
23
+ "reduce_sum",
24
+ "scatter",
25
+ ]
build/torch212-cxx11-cu132-aarch64-linux/_mra_cuda_10efe7e.abi3.so ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ca541eeb0040b9aa5719644bf330e80e55e23f23d7ded03fd1f5c5bddcaabbc9
3
+ size 2770896
build/torch212-cxx11-cu132-aarch64-linux/_ops.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ from . import _mra_cuda_10efe7e
3
+ ops = torch.ops._mra_cuda_10efe7e
4
+
5
+ def add_op_namespace_prefix(op_name: str):
6
+ """
7
+ Prefix op by namespace.
8
+ """
9
+ return f"_mra_cuda_10efe7e::{op_name}"
build/torch212-cxx11-cu132-aarch64-linux/metadata.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "name": "mra",
3
+ "id": "_mra_cuda_10efe7e",
4
+ "version": 1,
5
+ "license": "Apache-2.0",
6
+ "python-depends": [],
7
+ "backend": {
8
+ "type": "cuda",
9
+ "archs": [
10
+ "10.0",
11
+ "11.0",
12
+ "12.0+PTX",
13
+ "7.5",
14
+ "8.0",
15
+ "8.6",
16
+ "8.7",
17
+ "8.9",
18
+ "9.0"
19
+ ]
20
+ }
21
+ }
build/torch212-cxx11-cu132-aarch64-linux/mra/__init__.py ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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")))