@@ -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,
|
||||
|
||||
@@ -39,6 +39,7 @@ cache = dict()
|
||||
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
|
||||
@@ -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
|
||||
@@ -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:
|
||||
|
||||
Ссылка в новой задаче
Block a user