FP8 roofline support (#592)

Adding FP8 datatype to roofline feature in rocprof-compute on MI300-based systems.
FP8 now shows in terminal output and roofline csv, and outputs a standalone PDF.

---------

Signed-off-by: Carrie Fallows <Carrie.Fallows@amd.com>
This commit is contained in:
cfallows-amd
2025-03-07 02:54:50 -05:00
committed by Vignesh Edithal
parent f64c83fc5e
commit 848fa1dc18
8 changed files with 42 additions and 18 deletions
+13
View File
@@ -130,6 +130,9 @@ class Roofline:
dtype="I8",
fig=fp16_fig,
)
if self.__mspec.gpu_series != "MI200":
fig_fp8 = self.generate_plot(dtype="FP8")
# Create a legend and distinct kernel markers. This can be saved, optionally
self.__figure = go.Figure(
go.Scatter(
@@ -163,6 +166,11 @@ class Roofline:
self.__run_parameters["workload_dir"]
+ "/empirRoof_gpu-{}_int8_fp16.pdf".format(dev_id)
)
if self.__mspec.gpu_series != "MI200":
fig_fp8.write_image(
self.__run_parameters["workload_dir"]
+ "/empirRoof_gpu-{}_fp8.pdf".format(dev_id)
)
# only save a legend if kernel_names option is toggled
if self.__run_parameters["include_kernel_names"]:
self.__figure.write_image(
@@ -178,6 +186,11 @@ class Roofline:
self.__run_parameters["workload_dir"]
+ "/empirRoof_gpu-{}_int8_fp16.pdf".format(dev_id)
)
if self.__mspec.gpu_series != "MI200":
fig_fp8.write_image(
self.__run_parameters["workload_dir"]
+ "/empirRoof_gpu-{}_fp8.pdf".format(dev_id)
)
if self.__run_parameters["include_kernel_names"]:
self.__figure.write_image(
self.__run_parameters["workload_dir"] + "/kernelName_legend.pdf"