Skip to content

Commit 9778bb1

Browse files
committed
Update run_benchmarks.py
1 parent ebdecd4 commit 9778bb1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/mlx_metal_kernel_opt/run_benchmarks.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -133,7 +133,7 @@ def run_optimized_benchmark(args, original_dir):
133133
return None
134134

135135
# Apply the custom attention hook
136-
apply_hook, remove_hook = best_program.create_qwen3_optimization_hook()
136+
apply_hook, remove_hook = best_program.create_metal_qwen3_optimization_hook()
137137
print("🔧 Applying optimized attention hook...")
138138

139139
original_attention = apply_hook()

0 commit comments

Comments
 (0)