ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update __init__.py

Browse source
This commit is contained in:
Daniel Han 2025-01-22 16:46:54 -08:00
commit 78b756fd6e
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
unsloth/__init__.py
View file

@ -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()