Upload folder using huggingface_hub
Browse files
build/torch-universal/triton_kernels/__init__.py
CHANGED
|
@@ -1,3 +0,0 @@
|
|
| 1 |
-
from . import matmul_ogs
|
| 2 |
-
|
| 3 |
-
__all__ = ["matmul_ogs"]
|
|
|
|
|
|
|
|
|
|
|
|
build/torch-universal/triton_kernels/__pycache__/__init__.cpython-312.pyc
CHANGED
|
Binary files a/build/torch-universal/triton_kernels/__pycache__/__init__.cpython-312.pyc and b/build/torch-universal/triton_kernels/__pycache__/__init__.cpython-312.pyc differ
|
|
|
torch-ext/triton_kernels/matmul_ogs_details/_p_matmul_ogs.py
CHANGED
|
@@ -3,7 +3,7 @@
|
|
| 3 |
import torch
|
| 4 |
import triton
|
| 5 |
import triton.language as tl
|
| 6 |
-
from
|
| 7 |
from ..tensor_details.layout_details.blackwell_scale import unswizzle_mx_scale_bw
|
| 8 |
from ..numerics_details.flexpoint import (
|
| 9 |
float_to_flex,
|
|
|
|
| 3 |
import torch
|
| 4 |
import triton
|
| 5 |
import triton.language as tl
|
| 6 |
+
from .. import target_info
|
| 7 |
from ..tensor_details.layout_details.blackwell_scale import unswizzle_mx_scale_bw
|
| 8 |
from ..numerics_details.flexpoint import (
|
| 9 |
float_to_flex,
|