From 72031c8ae63f2a4f53f252f401024bfb53e760c0 Mon Sep 17 00:00:00 2001 From: coleramos425 Date: Mon, 13 Mar 2023 13:48:54 -0500 Subject: [PATCH] Comply to Python formatting Signed-off-by: coleramos425 --- src/omniperf_analyze/utils/roofline_calc.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/src/omniperf_analyze/utils/roofline_calc.py b/src/omniperf_analyze/utils/roofline_calc.py index 1fbca73e02..2750052337 100644 --- a/src/omniperf_analyze/utils/roofline_calc.py +++ b/src/omniperf_analyze/utils/roofline_calc.py @@ -218,10 +218,10 @@ def plot_application(sortType, ret_df, verbose): # Calculate + append AI data if # a) current KernelName is different than previous OR # b) We've reached the end of list - if(idx + 1 == df.shape[0]): + if idx + 1 == df.shape[0]: at_end = True else: - next_kernelName = df["KernelName"][idx+1] + next_kernelName = df["KernelName"][idx + 1] kernelName = df["KernelName"][idx] try: @@ -304,7 +304,9 @@ def plot_application(sortType, ret_df, verbose): try: lds_data += ( - (df["SQ_LDS_IDX_ACTIVE"][idx] - df["SQ_LDS_BANK_CONFLICT"][idx]) * 4 * L2_BANKS + (df["SQ_LDS_IDX_ACTIVE"][idx] - df["SQ_LDS_BANK_CONFLICT"][idx]) + * 4 + * L2_BANKS ) # L2_BANKS = 32 (since assuming mi200) except KeyError: if verbose >= 3: @@ -347,7 +349,7 @@ def plot_application(sortType, ret_df, verbose): calls += 1 - if sortType == "kernels" and (at_end == True or (kernelName != next_kernelName)): + if sortType == "kernels" and (at_end == True or (kernelName != next_kernelName)): myList.append( AI_Data( kernelName,