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:
committed by
Vignesh Edithal
parent
f64c83fc5e
commit
848fa1dc18
@@ -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"
|
||||
|
||||
Reference in New Issue
Block a user