Signed-off-by: josantos <josantos@amd.com>
Этот коммит содержится в:
josantos
2023-08-10 11:16:50 -05:00
родитель 59d77f9d81
Коммит bbb254ef43
2 изменённых файлов: 21 добавлений и 17 удалений
+12 -10
Просмотреть файл
@@ -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: