Kernels
danieldk HF Staff commited on
Commit
2e121aa
·
verified ·
1 Parent(s): d89fa00

Build uploaded using `kernels`.

Browse files
Files changed (44) hide show
  1. .gitattributes +9 -0
  2. build/torch210-cxx11-cpu-aarch64-linux/_ops.py +3 -3
  3. build/torch210-cxx11-cpu-aarch64-linux/{_relu_cpu_0a30093.abi3.so → _relu_cpu_86608d6.abi3.so} +1 -1
  4. build/torch210-cxx11-cpu-aarch64-linux/metadata.json +4 -1
  5. build/torch210-cxx11-cpu-aarch64-linux/relu/__init__.py +2 -2
  6. build/torch210-cxx11-cu126-aarch64-linux/_ops.py +3 -3
  7. build/torch210-cxx11-cu126-aarch64-linux/{_relu_cuda_0a30093.abi3.so → _relu_cuda_86608d6.abi3.so} +1 -1
  8. build/torch210-cxx11-cu126-aarch64-linux/relu/__init__.py +2 -2
  9. build/torch210-cxx11-cu128-aarch64-linux/_ops.py +3 -3
  10. build/torch210-cxx11-cu128-aarch64-linux/{_relu_cuda_0a30093.abi3.so → _relu_cuda_86608d6.abi3.so} +1 -1
  11. build/torch210-cxx11-cu128-aarch64-linux/relu/__init__.py +2 -2
  12. build/torch210-cxx11-cu130-aarch64-linux/_ops.py +3 -3
  13. build/torch210-cxx11-cu130-aarch64-linux/{_relu_cuda_0a30093.abi3.so → _relu_cuda_86608d6.abi3.so} +1 -1
  14. build/torch210-cxx11-cu130-aarch64-linux/relu/__init__.py +2 -2
  15. build/torch211-cxx11-cpu-aarch64-linux/__init__.py +16 -0
  16. build/torch211-cxx11-cpu-aarch64-linux/_ops.py +9 -0
  17. build/torch211-cxx11-cpu-aarch64-linux/_relu_cpu_86608d6.abi3.so +3 -0
  18. build/torch211-cxx11-cpu-aarch64-linux/layers/__init__.py +11 -0
  19. build/torch211-cxx11-cpu-aarch64-linux/metadata.json +8 -0
  20. build/torch211-cxx11-cpu-aarch64-linux/relu/__init__.py +26 -0
  21. build/torch211-cxx11-cu126-aarch64-linux/__init__.py +16 -0
  22. build/torch211-cxx11-cu126-aarch64-linux/_ops.py +9 -0
  23. build/torch211-cxx11-cu126-aarch64-linux/_relu_cuda_86608d6.abi3.so +3 -0
  24. build/torch211-cxx11-cu126-aarch64-linux/layers/__init__.py +11 -0
  25. build/torch211-cxx11-cu126-aarch64-linux/metadata.json +18 -0
  26. build/torch211-cxx11-cu126-aarch64-linux/relu/__init__.py +26 -0
  27. build/torch211-cxx11-cu128-aarch64-linux/__init__.py +16 -0
  28. build/torch211-cxx11-cu128-aarch64-linux/_ops.py +9 -0
  29. build/torch211-cxx11-cu128-aarch64-linux/_relu_cuda_86608d6.abi3.so +3 -0
  30. build/torch211-cxx11-cu128-aarch64-linux/layers/__init__.py +11 -0
  31. build/torch211-cxx11-cu128-aarch64-linux/metadata.json +21 -0
  32. build/torch211-cxx11-cu128-aarch64-linux/relu/__init__.py +26 -0
  33. build/torch211-cxx11-cu130-aarch64-linux/__init__.py +16 -0
  34. build/torch211-cxx11-cu130-aarch64-linux/_ops.py +9 -0
  35. build/torch211-cxx11-cu130-aarch64-linux/_relu_cuda_86608d6.abi3.so +3 -0
  36. build/torch211-cxx11-cu130-aarch64-linux/layers/__init__.py +11 -0
  37. build/torch211-cxx11-cu130-aarch64-linux/metadata.json +19 -0
  38. build/torch211-cxx11-cu130-aarch64-linux/relu/__init__.py +26 -0
  39. build/torch29-cxx11-cu129-aarch64-linux/__init__.py +16 -0
  40. build/torch29-cxx11-cu129-aarch64-linux/_ops.py +9 -0
  41. build/torch29-cxx11-cu129-aarch64-linux/_relu_cuda_86608d6.abi3.so +3 -0
  42. build/torch29-cxx11-cu129-aarch64-linux/layers/__init__.py +11 -0
  43. build/torch29-cxx11-cu129-aarch64-linux/metadata.json +21 -0
  44. build/torch29-cxx11-cu129-aarch64-linux/relu/__init__.py +26 -0
.gitattributes CHANGED
@@ -284,3 +284,12 @@ build/torch29-cxx11-cu128-aarch64-linux/_relu_cuda_0a30093.abi3.so filter=lfs di
284
  build/torch29-cxx11-cu130-aarch64-linux/_relu_cuda_0a30093.abi3.so filter=lfs diff=lfs merge=lfs -text
285
  build/torch210-cu128-x86_64-windows/_relu_cuda_fd389c8.pyd filter=lfs diff=lfs merge=lfs -text
286
  build/torch210-xpu20253-x86_64-windows/_relu_xpu_fd389c8.pyd filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
 
 
 
 
 
284
  build/torch29-cxx11-cu130-aarch64-linux/_relu_cuda_0a30093.abi3.so filter=lfs diff=lfs merge=lfs -text
285
  build/torch210-cu128-x86_64-windows/_relu_cuda_fd389c8.pyd filter=lfs diff=lfs merge=lfs -text
286
  build/torch210-xpu20253-x86_64-windows/_relu_xpu_fd389c8.pyd filter=lfs diff=lfs merge=lfs -text
287
+ build/torch210-cxx11-cpu-aarch64-linux/_relu_cpu_86608d6.abi3.so filter=lfs diff=lfs merge=lfs -text
288
+ build/torch210-cxx11-cu126-aarch64-linux/_relu_cuda_86608d6.abi3.so filter=lfs diff=lfs merge=lfs -text
289
+ build/torch210-cxx11-cu128-aarch64-linux/_relu_cuda_86608d6.abi3.so filter=lfs diff=lfs merge=lfs -text
290
+ build/torch210-cxx11-cu130-aarch64-linux/_relu_cuda_86608d6.abi3.so filter=lfs diff=lfs merge=lfs -text
291
+ build/torch211-cxx11-cpu-aarch64-linux/_relu_cpu_86608d6.abi3.so filter=lfs diff=lfs merge=lfs -text
292
+ build/torch211-cxx11-cu126-aarch64-linux/_relu_cuda_86608d6.abi3.so filter=lfs diff=lfs merge=lfs -text
293
+ build/torch211-cxx11-cu128-aarch64-linux/_relu_cuda_86608d6.abi3.so filter=lfs diff=lfs merge=lfs -text
294
+ build/torch211-cxx11-cu130-aarch64-linux/_relu_cuda_86608d6.abi3.so filter=lfs diff=lfs merge=lfs -text
295
+ build/torch29-cxx11-cu129-aarch64-linux/_relu_cuda_86608d6.abi3.so filter=lfs diff=lfs merge=lfs -text
build/torch210-cxx11-cpu-aarch64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _relu_cpu_0a30093
3
- ops = torch.ops._relu_cpu_0a30093
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_relu_cpu_0a30093::{op_name}"
 
1
  import torch
2
+ from . import _relu_cpu_86608d6
3
+ ops = torch.ops._relu_cpu_86608d6
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_relu_cpu_86608d6::{op_name}"
build/torch210-cxx11-cpu-aarch64-linux/{_relu_cpu_0a30093.abi3.so → _relu_cpu_86608d6.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8d7a2710d50ffac544ab37f351f4e9808743a46d851014bfec55032fe2b772c5
3
  size 2025352
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:596435958298e61250b099f9bdf79b5f2436875a1de8e9a4cc03f9d98cb42f82
3
  size 2025352
build/torch210-cxx11-cpu-aarch64-linux/metadata.json CHANGED
@@ -1,5 +1,8 @@
1
  {
2
  "version": 1,
3
  "license": "Apache-2.0",
4
- "python-depends": []
 
 
 
5
  }
 
1
  {
2
  "version": 1,
3
  "license": "Apache-2.0",
4
+ "python-depends": [],
5
+ "backend": {
6
+ "type": "cpu"
7
+ }
8
  }
build/torch210-cxx11-cpu-aarch64-linux/relu/__init__.py CHANGED
@@ -1,10 +1,10 @@
1
  import ctypes
 
2
  import sys
3
-
4
- import importlib
5
  from pathlib import Path
6
  from types import ModuleType
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
 
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
build/torch210-cxx11-cu126-aarch64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _relu_cuda_0a30093
3
- ops = torch.ops._relu_cuda_0a30093
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_relu_cuda_0a30093::{op_name}"
 
1
  import torch
2
+ from . import _relu_cuda_86608d6
3
+ ops = torch.ops._relu_cuda_86608d6
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_relu_cuda_86608d6::{op_name}"
build/torch210-cxx11-cu126-aarch64-linux/{_relu_cuda_0a30093.abi3.so → _relu_cuda_86608d6.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8a2f67c7ae80600668fec37d2b55d226cd025380f398510de87ab070a5bedbf5
3
  size 2103912
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4409e4835a9af3abf816cd252c489a53ccfca967ad654544e97403ba75b339ea
3
  size 2103912
build/torch210-cxx11-cu126-aarch64-linux/relu/__init__.py CHANGED
@@ -1,10 +1,10 @@
1
  import ctypes
 
2
  import sys
3
-
4
- import importlib
5
  from pathlib import Path
6
  from types import ModuleType
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
 
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
build/torch210-cxx11-cu128-aarch64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _relu_cuda_0a30093
3
- ops = torch.ops._relu_cuda_0a30093
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_relu_cuda_0a30093::{op_name}"
 
1
  import torch
2
+ from . import _relu_cuda_86608d6
3
+ ops = torch.ops._relu_cuda_86608d6
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_relu_cuda_86608d6::{op_name}"
build/torch210-cxx11-cu128-aarch64-linux/{_relu_cuda_0a30093.abi3.so → _relu_cuda_86608d6.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3fc87bfb8bbbc41d9d78c65959ef906e2b1940fd4c609ad1e45eb8e6d524ff5f
3
  size 2235240
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eb1d76a7970041e8b408d493adb9250bb755581255aeae7ba2cf44d22ae2490b
3
  size 2235240
build/torch210-cxx11-cu128-aarch64-linux/relu/__init__.py CHANGED
@@ -1,10 +1,10 @@
1
  import ctypes
 
2
  import sys
3
-
4
- import importlib
5
  from pathlib import Path
6
  from types import ModuleType
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
 
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
build/torch210-cxx11-cu130-aarch64-linux/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _relu_cuda_0a30093
3
- ops = torch.ops._relu_cuda_0a30093
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_relu_cuda_0a30093::{op_name}"
 
1
  import torch
2
+ from . import _relu_cuda_86608d6
3
+ ops = torch.ops._relu_cuda_86608d6
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_relu_cuda_86608d6::{op_name}"
build/torch210-cxx11-cu130-aarch64-linux/{_relu_cuda_0a30093.abi3.so → _relu_cuda_86608d6.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f85bfa4560f261b932b0a983b6b8817391353f622fa06152b4794c58edaab738
3
  size 2236952
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:af6ab238580c7b5855200df6a0cac5dfa7fbd36342a2c82f26431d046ec7fddc
3
  size 2236952
build/torch210-cxx11-cu130-aarch64-linux/relu/__init__.py CHANGED
@@ -1,10 +1,10 @@
1
  import ctypes
 
2
  import sys
3
-
4
- import importlib
5
  from pathlib import Path
6
  from types import ModuleType
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
 
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
build/torch211-cxx11-cpu-aarch64-linux/__init__.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from typing import Optional
2
+
3
+ import torch
4
+
5
+ from ._ops import ops
6
+
7
+ from . import layers
8
+
9
+
10
+ def relu(x: torch.Tensor, out: Optional[torch.Tensor] = None) -> torch.Tensor:
11
+ if out is None:
12
+ out = torch.empty_like(x)
13
+ ops.relu(out, x)
14
+ return out
15
+
16
+ __all__ = ["relu", "layers"]
build/torch211-cxx11-cpu-aarch64-linux/_ops.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ from . import _relu_cpu_86608d6
3
+ ops = torch.ops._relu_cpu_86608d6
4
+
5
+ def add_op_namespace_prefix(op_name: str):
6
+ """
7
+ Prefix op by namespace.
8
+ """
9
+ return f"_relu_cpu_86608d6::{op_name}"
build/torch211-cxx11-cpu-aarch64-linux/_relu_cpu_86608d6.abi3.so ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4475353646ad4a7adc698f5abc6f3c053c3deaaffd980710e31fddddc65de8a2
3
+ size 2025352
build/torch211-cxx11-cpu-aarch64-linux/layers/__init__.py ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ import torch.nn as nn
3
+
4
+ from .._ops import ops
5
+
6
+
7
+ class ReLU(nn.Module):
8
+ def forward(self, x: torch.Tensor) -> torch.Tensor:
9
+ out = torch.empty_like(x)
10
+ ops.relu(out, x)
11
+ return out
build/torch211-cxx11-cpu-aarch64-linux/metadata.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "version": 1,
3
+ "license": "Apache-2.0",
4
+ "python-depends": [],
5
+ "backend": {
6
+ "type": "cpu"
7
+ }
8
+ }
build/torch211-cxx11-cpu-aarch64-linux/relu/__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/torch211-cxx11-cu126-aarch64-linux/__init__.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from typing import Optional
2
+
3
+ import torch
4
+
5
+ from ._ops import ops
6
+
7
+ from . import layers
8
+
9
+
10
+ def relu(x: torch.Tensor, out: Optional[torch.Tensor] = None) -> torch.Tensor:
11
+ if out is None:
12
+ out = torch.empty_like(x)
13
+ ops.relu(out, x)
14
+ return out
15
+
16
+ __all__ = ["relu", "layers"]
build/torch211-cxx11-cu126-aarch64-linux/_ops.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ from . import _relu_cuda_86608d6
3
+ ops = torch.ops._relu_cuda_86608d6
4
+
5
+ def add_op_namespace_prefix(op_name: str):
6
+ """
7
+ Prefix op by namespace.
8
+ """
9
+ return f"_relu_cuda_86608d6::{op_name}"
build/torch211-cxx11-cu126-aarch64-linux/_relu_cuda_86608d6.abi3.so ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d4f3cb1b7d262aa66d86d0a57521afbf6825ef2be6ebe0c294c48a28c792e293
3
+ size 2100128
build/torch211-cxx11-cu126-aarch64-linux/layers/__init__.py ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ import torch.nn as nn
3
+
4
+ from .._ops import ops
5
+
6
+
7
+ class ReLU(nn.Module):
8
+ def forward(self, x: torch.Tensor) -> torch.Tensor:
9
+ out = torch.empty_like(x)
10
+ ops.relu(out, x)
11
+ return out
build/torch211-cxx11-cu126-aarch64-linux/metadata.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "version": 1,
3
+ "license": "Apache-2.0",
4
+ "python-depends": [],
5
+ "backend": {
6
+ "type": "cuda",
7
+ "archs": [
8
+ "7.0",
9
+ "7.2",
10
+ "7.5",
11
+ "8.0",
12
+ "8.6",
13
+ "8.7",
14
+ "8.9",
15
+ "9.0+PTX"
16
+ ]
17
+ }
18
+ }
build/torch211-cxx11-cu126-aarch64-linux/relu/__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/torch211-cxx11-cu128-aarch64-linux/__init__.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from typing import Optional
2
+
3
+ import torch
4
+
5
+ from ._ops import ops
6
+
7
+ from . import layers
8
+
9
+
10
+ def relu(x: torch.Tensor, out: Optional[torch.Tensor] = None) -> torch.Tensor:
11
+ if out is None:
12
+ out = torch.empty_like(x)
13
+ ops.relu(out, x)
14
+ return out
15
+
16
+ __all__ = ["relu", "layers"]
build/torch211-cxx11-cu128-aarch64-linux/_ops.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ from . import _relu_cuda_86608d6
3
+ ops = torch.ops._relu_cuda_86608d6
4
+
5
+ def add_op_namespace_prefix(op_name: str):
6
+ """
7
+ Prefix op by namespace.
8
+ """
9
+ return f"_relu_cuda_86608d6::{op_name}"
build/torch211-cxx11-cu128-aarch64-linux/_relu_cuda_86608d6.abi3.so ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:74288ac0fda279a679e985ed3f184dada9eb3769283d3fe546eb05f1628e5f57
3
+ size 2231464
build/torch211-cxx11-cu128-aarch64-linux/layers/__init__.py ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ import torch.nn as nn
3
+
4
+ from .._ops import ops
5
+
6
+
7
+ class ReLU(nn.Module):
8
+ def forward(self, x: torch.Tensor) -> torch.Tensor:
9
+ out = torch.empty_like(x)
10
+ ops.relu(out, x)
11
+ return out
build/torch211-cxx11-cu128-aarch64-linux/metadata.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "version": 1,
3
+ "license": "Apache-2.0",
4
+ "python-depends": [],
5
+ "backend": {
6
+ "type": "cuda",
7
+ "archs": [
8
+ "10.0",
9
+ "10.1",
10
+ "12.0+PTX",
11
+ "7.0",
12
+ "7.2",
13
+ "7.5",
14
+ "8.0",
15
+ "8.6",
16
+ "8.7",
17
+ "8.9",
18
+ "9.0"
19
+ ]
20
+ }
21
+ }
build/torch211-cxx11-cu128-aarch64-linux/relu/__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/torch211-cxx11-cu130-aarch64-linux/__init__.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from typing import Optional
2
+
3
+ import torch
4
+
5
+ from ._ops import ops
6
+
7
+ from . import layers
8
+
9
+
10
+ def relu(x: torch.Tensor, out: Optional[torch.Tensor] = None) -> torch.Tensor:
11
+ if out is None:
12
+ out = torch.empty_like(x)
13
+ ops.relu(out, x)
14
+ return out
15
+
16
+ __all__ = ["relu", "layers"]
build/torch211-cxx11-cu130-aarch64-linux/_ops.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ from . import _relu_cuda_86608d6
3
+ ops = torch.ops._relu_cuda_86608d6
4
+
5
+ def add_op_namespace_prefix(op_name: str):
6
+ """
7
+ Prefix op by namespace.
8
+ """
9
+ return f"_relu_cuda_86608d6::{op_name}"
build/torch211-cxx11-cu130-aarch64-linux/_relu_cuda_86608d6.abi3.so ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:67ff85126ba3ae10041020f21b1f016dd0f59a62ea4f8d0687af55400a195d8d
3
+ size 2233176
build/torch211-cxx11-cu130-aarch64-linux/layers/__init__.py ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ import torch.nn as nn
3
+
4
+ from .._ops import ops
5
+
6
+
7
+ class ReLU(nn.Module):
8
+ def forward(self, x: torch.Tensor) -> torch.Tensor:
9
+ out = torch.empty_like(x)
10
+ ops.relu(out, x)
11
+ return out
build/torch211-cxx11-cu130-aarch64-linux/metadata.json ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "version": 1,
3
+ "license": "Apache-2.0",
4
+ "python-depends": [],
5
+ "backend": {
6
+ "type": "cuda",
7
+ "archs": [
8
+ "10.0",
9
+ "11.0",
10
+ "12.0+PTX",
11
+ "7.5",
12
+ "8.0",
13
+ "8.6",
14
+ "8.7",
15
+ "8.9",
16
+ "9.0"
17
+ ]
18
+ }
19
+ }
build/torch211-cxx11-cu130-aarch64-linux/relu/__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/torch29-cxx11-cu129-aarch64-linux/__init__.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from typing import Optional
2
+
3
+ import torch
4
+
5
+ from ._ops import ops
6
+
7
+ from . import layers
8
+
9
+
10
+ def relu(x: torch.Tensor, out: Optional[torch.Tensor] = None) -> torch.Tensor:
11
+ if out is None:
12
+ out = torch.empty_like(x)
13
+ ops.relu(out, x)
14
+ return out
15
+
16
+ __all__ = ["relu", "layers"]
build/torch29-cxx11-cu129-aarch64-linux/_ops.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ from . import _relu_cuda_86608d6
3
+ ops = torch.ops._relu_cuda_86608d6
4
+
5
+ def add_op_namespace_prefix(op_name: str):
6
+ """
7
+ Prefix op by namespace.
8
+ """
9
+ return f"_relu_cuda_86608d6::{op_name}"
build/torch29-cxx11-cu129-aarch64-linux/_relu_cuda_86608d6.abi3.so ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e48906b3c3019986065ae16c2a7023cb30ab15b896a247f17b55657fda076423
3
+ size 2234392
build/torch29-cxx11-cu129-aarch64-linux/layers/__init__.py ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ import torch.nn as nn
3
+
4
+ from .._ops import ops
5
+
6
+
7
+ class ReLU(nn.Module):
8
+ def forward(self, x: torch.Tensor) -> torch.Tensor:
9
+ out = torch.empty_like(x)
10
+ ops.relu(out, x)
11
+ return out
build/torch29-cxx11-cu129-aarch64-linux/metadata.json ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "version": 1,
3
+ "license": "Apache-2.0",
4
+ "python-depends": [],
5
+ "backend": {
6
+ "type": "cuda",
7
+ "archs": [
8
+ "10.0",
9
+ "10.1",
10
+ "12.0+PTX",
11
+ "7.0",
12
+ "7.2",
13
+ "7.5",
14
+ "8.0",
15
+ "8.6",
16
+ "8.7",
17
+ "8.9",
18
+ "9.0"
19
+ ]
20
+ }
21
+ }
build/torch29-cxx11-cu129-aarch64-linux/relu/__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")))