Skip to content

Commit ced0809

Browse files
committed
Add nfsstore bandwidth testing script
1 parent 3dd93f8 commit ced0809

File tree

1 file changed

+10
-3
lines changed

1 file changed

+10
-3
lines changed

ucm/store/test/e2e/nfsstore_embed_fetch_run.py

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,14 @@ def main():
7676
io_num = int(num_tokens / block_len * block_layer)
7777

7878
# Run test and get results
79-
avg_w_size, avg_w_time, avg_w_bw, avg_r_time, avg_r_bw, avg_r_size = run(
79+
(
80+
avg_w_size,
81+
avg_w_time,
82+
avg_w_bw,
83+
avg_r_time,
84+
avg_r_bw,
85+
avg_r_size,
86+
) = run(
8087
storage_backends,
8188
device_id,
8289
repeat,
@@ -86,7 +93,7 @@ def main():
8693
num_tokens,
8794
block_layer,
8895
head_size,
89-
block_elem_size
96+
block_elem_size,
9097
)
9198

9299
writer.writerow(
@@ -117,4 +124,4 @@ def main():
117124

118125
if __name__ == "__main__":
119126
os.environ["UC_LOGGER_LEVEL"] = "debug"
120-
main()
127+
main()

0 commit comments

Comments
 (0)