Merge pull request #13042 from wilsonwang371/no-hardcoded-repeat

tools: no hardcoded repeat times in rw-benchmark scripts
This commit is contained in:
Piotr Tabor 2021-05-26 08:36:10 +02:00 committed by GitHub
commit db556c11b2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 7 deletions

View File

@ -43,7 +43,8 @@ def load_data_files(*args):
try:
for df in df_list:
new_df = df[['ratio', 'conn_size', 'value_size']].copy()
tmp = [df[x].str.split(':') for x in ['1', '2', '3', '4', '5']]
cols = [x for x in df.columns if x.find('iter') != -1]
tmp = [df[x].str.split(':') for x in cols]
read_df = [x.apply(lambda x: float(x[0])) for x in tmp]
read_avg = sum(read_df)/len(read_df)
@ -73,7 +74,7 @@ def plot_data(title, *args):
count += 1
plt.subplot(4, 2, count)
plt.tripcolor(df['conn_size'], df['value_size'], df['read'] + df['write'])
plt.title('R/W Ratio {:.2f}'.format(val))
plt.title('R/W Ratio {:.4f}'.format(val))
plt.yscale('log', base=2)
plt.ylabel('Value Size')
plt.xscale('log', base=2)
@ -99,13 +100,13 @@ def plot_data(title, *args):
cmap_name = 'viridis'
plt.tripcolor(df['conn_size'], df['value_size'], df['read'] + df['write'], cmap=plt.get_cmap(cmap_name))
if count2 == 1:
plt.title('{}\nR/W Ratio {:.2f}'.format(os.path.basename(params.input_file_a), val))
plt.title('{}\nR/W Ratio {:.4f}'.format(os.path.basename(params.input_file_a), val))
elif count2 == 2:
plt.title('{}\nR/W Ratio {:.2f}'.format(os.path.basename(params.input_file_b), val))
plt.title('{}\nR/W Ratio {:.4f}'.format(os.path.basename(params.input_file_b), val))
elif count2 == 3:
plt.title('Delta\nR/W Ratio {:.2f}'.format(val))
plt.title('Delta\nR/W Ratio {:.4f}'.format(val))
else:
plt.title('R/W Ratio {:.2f}'.format(val))
plt.title('R/W Ratio {:.4f}'.format(val))
plt.yscale('log', base=2)
plt.ylabel('Value Size')
plt.xscale('log', base=2)

View File

@ -48,7 +48,7 @@ function check_prerequisite() {
exit 1
fi
cat >"${OUTPUT_FILE}" <<EOF
ratio, conn_size, value_size, 1, 2, 3, 4, 5
ratio,conn_size,value_size$(for i in $(seq 1 ${REPEAT_COUNT});do echo -n ",iter$i"; done)
EOF
}