Hugging Face's logo

Kernels:
kernels-community
/
trimul-gpumode
Trusted publisher

Kernel card Files
xet
 Community
kernels-bot commited on
Commit
5b8d6ad
·
verified ·
1 Parent(s): 3bab7c3

Uploaded using `kernel-builder`.

Browse files
Files changed (6) hide show
  1. build/torch-cuda/_ops.py +33 -3
  2. build/torch-cuda/metadata.json +3 -0
  3. build/torch-rocm/_ops.py +33 -3
  4. build/torch-rocm/metadata.json +3 -0
  5. build/torch-xpu/_ops.py +33 -3
  6. build/torch-xpu/metadata.json +3 -0
build/torch-cuda/_ops.py CHANGED
@@ -1,8 +1,38 @@
1
  import torch
2
- ops = torch.ops._trimul_gpumode_a721306
3
 
4
- def add_op_namespace_prefix(op_name: str):
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5
  """
6
  Prefix op by namespace.
7
  """
8
- return f"_trimul_gpumode_a721306::{op_name}"
 
1
  import torch
 
2
 
3
+ def get_backend() -> str:
4
+ """Detect the backend by inspecting torch."""
5
+ import torch
6
+
7
+ if hasattr(torch, "neuron"):
8
+ # Needs to be sorted before specific Torch builds, since Neuron
9
+ # extension can be loaded into e.g. CUDA Torch builds.
10
+ return "neuron"
11
+ elif torch.version.cuda is not None:
12
+ return "cuda"
13
+ elif torch.version.hip is not None:
14
+ return "rocm"
15
+ elif torch.backends.mps.is_available():
16
+ return "metal"
17
+ elif hasattr(torch.version, "xpu") and torch.version.xpu is not None:
18
+ return "xpu"
19
+ else:
20
+ return "cpu"
21
+
22
+
23
+ def _find_ops_name() -> str:
24
+ kernel_name = "trimul_gpumode"
25
+ unique_id = "86f75d9"
26
+ backend = get_backend()
27
+ return f"_{kernel_name}_{backend}_{unique_id}"
28
+
29
+
30
+ _OPS_NAME = _find_ops_name()
31
+
32
+ ops = getattr(torch.ops, _OPS_NAME)
33
+
34
+ def add_op_namespace_prefix(op_name: str) -> str:
35
  """
36
  Prefix op by namespace.
37
  """
38
+ return f"{_OPS_NAME}::{op_name}"
build/torch-cuda/metadata.json CHANGED
@@ -1,5 +1,8 @@
1
  {
 
 
2
  "version": 1,
 
3
  "python-depends": [],
4
  "backend": {
5
  "type": "cuda"
 
1
  {
2
+ "name": "trimul-gpumode",
3
+ "id": "_trimul_gpumode_cuda_86f75d9",
4
  "version": 1,
5
+ "license": "Apache-2.0",
6
  "python-depends": [],
7
  "backend": {
8
  "type": "cuda"
build/torch-rocm/_ops.py CHANGED
@@ -1,8 +1,38 @@
1
  import torch
2
- ops = torch.ops._trimul_gpumode_a721306
3
 
4
- def add_op_namespace_prefix(op_name: str):
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5
  """
6
  Prefix op by namespace.
7
  """
8
- return f"_trimul_gpumode_a721306::{op_name}"
 
1
  import torch
 
2
 
3
+ def get_backend() -> str:
4
+ """Detect the backend by inspecting torch."""
5
+ import torch
6
+
7
+ if hasattr(torch, "neuron"):
8
+ # Needs to be sorted before specific Torch builds, since Neuron
9
+ # extension can be loaded into e.g. CUDA Torch builds.
10
+ return "neuron"
11
+ elif torch.version.cuda is not None:
12
+ return "cuda"
13
+ elif torch.version.hip is not None:
14
+ return "rocm"
15
+ elif torch.backends.mps.is_available():
16
+ return "metal"
17
+ elif hasattr(torch.version, "xpu") and torch.version.xpu is not None:
18
+ return "xpu"
19
+ else:
20
+ return "cpu"
21
+
22
+
23
+ def _find_ops_name() -> str:
24
+ kernel_name = "trimul_gpumode"
25
+ unique_id = "86f75d9"
26
+ backend = get_backend()
27
+ return f"_{kernel_name}_{backend}_{unique_id}"
28
+
29
+
30
+ _OPS_NAME = _find_ops_name()
31
+
32
+ ops = getattr(torch.ops, _OPS_NAME)
33
+
34
+ def add_op_namespace_prefix(op_name: str) -> str:
35
  """
36
  Prefix op by namespace.
37
  """
38
+ return f"{_OPS_NAME}::{op_name}"
build/torch-rocm/metadata.json CHANGED
@@ -1,5 +1,8 @@
1
  {
 
 
2
  "version": 1,
 
3
  "python-depends": [],
4
  "backend": {
5
  "type": "rocm"
 
1
  {
2
+ "name": "trimul-gpumode",
3
+ "id": "_trimul_gpumode_rocm_86f75d9",
4
  "version": 1,
5
+ "license": "Apache-2.0",
6
  "python-depends": [],
7
  "backend": {
8
  "type": "rocm"
build/torch-xpu/_ops.py CHANGED
@@ -1,8 +1,38 @@
1
  import torch
2
- ops = torch.ops._trimul_gpumode_a721306
3
 
4
- def add_op_namespace_prefix(op_name: str):
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5
  """
6
  Prefix op by namespace.
7
  """
8
- return f"_trimul_gpumode_a721306::{op_name}"
 
1
  import torch
 
2
 
3
+ def get_backend() -> str:
4
+ """Detect the backend by inspecting torch."""
5
+ import torch
6
+
7
+ if hasattr(torch, "neuron"):
8
+ # Needs to be sorted before specific Torch builds, since Neuron
9
+ # extension can be loaded into e.g. CUDA Torch builds.
10
+ return "neuron"
11
+ elif torch.version.cuda is not None:
12
+ return "cuda"
13
+ elif torch.version.hip is not None:
14
+ return "rocm"
15
+ elif torch.backends.mps.is_available():
16
+ return "metal"
17
+ elif hasattr(torch.version, "xpu") and torch.version.xpu is not None:
18
+ return "xpu"
19
+ else:
20
+ return "cpu"
21
+
22
+
23
+ def _find_ops_name() -> str:
24
+ kernel_name = "trimul_gpumode"
25
+ unique_id = "86f75d9"
26
+ backend = get_backend()
27
+ return f"_{kernel_name}_{backend}_{unique_id}"
28
+
29
+
30
+ _OPS_NAME = _find_ops_name()
31
+
32
+ ops = getattr(torch.ops, _OPS_NAME)
33
+
34
+ def add_op_namespace_prefix(op_name: str) -> str:
35
  """
36
  Prefix op by namespace.
37
  """
38
+ return f"{_OPS_NAME}::{op_name}"
build/torch-xpu/metadata.json CHANGED
@@ -1,5 +1,8 @@
1
  {
 
 
2
  "version": 1,
 
3
  "python-depends": [],
4
  "backend": {
5
  "type": "xpu"
 
1
  {
2
+ "name": "trimul-gpumode",
3
+ "id": "_trimul_gpumode_xpu_86f75d9",
4
  "version": 1,
5
+ "license": "Apache-2.0",
6
  "python-depends": [],
7
  "backend": {
8
  "type": "xpu"