Skip to content

Commit 68aa0b9

Browse files
committed
fix: retry compiling with image
1 parent d169cca commit 68aa0b9

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

flash_attn/benchmark.md

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -375,7 +375,5 @@ def benchmark_fn(seq_len: int, provider: str):
375375
with torch.inference_mode():
376376
correctness()
377377
fig = benchmark_fn.run(print_data=True, save_path=output_dir.as_posix())
378-
fig.savefig(output_dir / "attention_benchmark.png", dpi=300, bbox_inches="tight")
379-
print(f"Benchmark plot saved to: {output_dir / 'attention_benchmark.png'}")
380378

381379
```

0 commit comments

Comments
 (0)