From 78b756fd6e91ff20bf3ad1759de614efb0de2b2d Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 22 Jan 2025 16:46:54 -0800 Subject: [PATCH] Update __init__.py --- unsloth/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/unsloth/__init__.py b/unsloth/__init__.py index 8c949b0585..9068df99b4 100644 --- a/unsloth/__init__.py +++ b/unsloth/__init__.py @@ -153,8 +153,8 @@ def fix_triton_ops(): early_config_prune, estimate_matmul_time, ) - triton.ops.early_config_prune = early_config_prune - triton.ops.estimate_matmul_time = 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 pass pass fix_triton_ops()