From efd538270817c339b80457311fa8bbe22dbe5b82 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Wed, 22 Jan 2025 16:53:35 -0800
Subject: [PATCH] triton.ops error

---
 unsloth/__init__.py                        | 22 +++++++++++++++++-----
 unsloth/{kernels => }/matmul_perf_model.py |  0
 2 files changed, 17 insertions(+), 5 deletions(-)
 rename unsloth/{kernels => }/matmul_perf_model.py (100%)

diff --git a/unsloth/__init__.py b/unsloth/__init__.py
index 9068df99b4..74a0adce57 100644
--- a/unsloth/__init__.py
+++ b/unsloth/__init__.py
@@ -149,12 +149,24 @@ def fix_triton_ops():
         import triton.ops
     except:
         # Triton 3.2 removed triton.ops
-        from .kernels.matmul_perf_model import (
-            early_config_prune,
-            estimate_matmul_time,
+        from .matmul_perf_model import (
+            early_config_prune   as _early_config_prune,
+            estimate_matmul_time as _estimate_matmul_time,
         )
-        triton.ops.matmul_perf_model.early_config_prune   = early_config_prune
-        triton.ops.matmul_perf_model.estimate_matmul_time = estimate_matmul_time
+        class PerfOps:
+            def __init__(self): return
+            @staticmethod
+            def early_config_prune(*args, **kwargs):
+                return _early_config_prune(*args, **kwargs)
+            @staticmethod
+            def estimate_matmul_time(*args, **kwargs):
+                return _estimate_matmul_time(*args, **kwargs)
+        pass
+        class TritonOps:
+            __slots__ = "matmul_perf_model",
+            def __init__(self): self.matmul_perf_model = PerfOps()
+        pass
+        triton.ops = TritonOps()
     pass
 pass
 fix_triton_ops()
diff --git a/unsloth/kernels/matmul_perf_model.py b/unsloth/matmul_perf_model.py
similarity index 100%
rename from unsloth/kernels/matmul_perf_model.py
rename to unsloth/matmul_perf_model.py