Skip to content

Commit

Permalink
reformatting
Browse files Browse the repository at this point in the history
Signed-off-by: josantos <josantos@amd.com>
  • Loading branch information
JoseSantosAMD committed Aug 10, 2023
1 parent 59d77f9 commit bbb254e
Show file tree
Hide file tree
Showing 2 changed files with 21 additions and 17 deletions.
16 changes: 9 additions & 7 deletions src/omniperf_analyze/omniperf_analyze.py
Original file line number Diff line number Diff line change
Expand Up @@ -222,15 +222,17 @@ def run_cli(args, runs):
# If we assume the panel layout for all archs are similar, it doesn't matter
# which archConfig passed into show_all function.
# After decide to how to manage kernels display patterns, we can revisit it.
cache =dict()
cache = dict()
for d in args.path:
#demangle
# demangle
for filename in os.listdir(d[0]):
if filename.endswith('.csv'):
df = pd.read_csv(os.path.join(d[0],filename))
new_df = csv_converter.kernel_name_shortener(df, cache, args.kernelVerbose)
new_df.to_csv(os.path.join(d[0],filename), index=False)

if filename.endswith(".csv"):
df = pd.read_csv(os.path.join(d[0], filename))
new_df = csv_converter.kernel_name_shortener(
df, cache, args.kernelVerbose
)
new_df.to_csv(os.path.join(d[0], filename), index=False)

file_io.create_df_kernel_top_stats(
d[0],
runs[d[0]].filter_gpu_ids,
Expand Down
22 changes: 12 additions & 10 deletions src/utils/csv_converter.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,7 @@
supported_arch = {"gfx906": "mi50", "gfx908": "mi100", "gfx90a": "mi200"}
MAX_SERVER_SEL_DELAY = 5000 # 5 sec connection timeout


def kernel_name_shortener(df, cache, level):
if level >= 5:
return df
Expand Down Expand Up @@ -116,6 +117,7 @@ def kernel_name_shortener(df, cache, level):

return df


# Verify target directory and setup connection
def parse(args, profileAndExport):
host = args.host
Expand Down Expand Up @@ -243,16 +245,16 @@ def convert_folder(connectionInfo, Extractionlvl):
# os.system(cmd)
# else:
cmd = (
"mongoimport --quiet --uri mongodb://{}:{}@{}:{}/{}?authSource=admin --file {} -c {} --drop --type csv --headerline"
).format(
connectionInfo["username"],
connectionInfo["password"],
connectionInfo["host"],
connectionInfo["port"],
connectionInfo["db"],
connectionInfo["workload"] + "/" + file,
fileName,
)
"mongoimport --quiet --uri mongodb://{}:{}@{}:{}/{}?authSource=admin --file {} -c {} --drop --type csv --headerline"
).format(
connectionInfo["username"],
connectionInfo["password"],
connectionInfo["host"],
connectionInfo["port"],
connectionInfo["db"],
connectionInfo["workload"] + "/" + file,
fileName,
)
os.system(cmd)
i += 1
except pd.errors.EmptyDataError:
Expand Down

0 comments on commit bbb254e

Please sign in to comment.