Skip to content

Commit

Permalink
better file name
Browse files Browse the repository at this point in the history
  • Loading branch information
drisspg committed Oct 10, 2024
1 parent 34e51d0 commit 3f7e8a9
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion benchmarks/fp8_matmul.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@
matmul_tma_persistent,
matmul_device_tma_persistent,
)
from datetime import datetime
from enum import Enum
import csv

Expand Down Expand Up @@ -242,7 +243,8 @@ def plot_tflops_comparison(df, save_path: Path):
plt.tight_layout()

# Generate the file name and save in the same directory as the CSV file
file_name = f"fp8_kernel_comparison_{m_value}_{n_value}.png"
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
file_name = f"fp8_kernel_comparison_{m_value}_{n_value}_{timestamp}.png"
graph_path = save_path.parent / file_name
plt.savefig(graph_path, dpi=300)
print(f"TFLOPS comparison plot saved as {graph_path}")
Expand Down

0 comments on commit 3f7e8a9

Please sign in to comment.