We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d169cca commit 68aa0b9Copy full SHA for 68aa0b9
flash_attn/benchmark.md
@@ -375,7 +375,5 @@ def benchmark_fn(seq_len: int, provider: str):
375
with torch.inference_mode():
376
correctness()
377
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'}")
380
381
```
0 commit comments