Add nfsstore bandwidth testing script

Author: zhou-haitao

ucm/store/test/e2e/nfsstore_embed_fetch_run.py

@@ -23,12 +23,21 @@
 # SOFTWARE.
 #
 import csv
+import multiprocessing
 import os
 from typing import List
 
 from nfsstore_embed_fetch import run
 
 
+def run_wrapper(result_queue, *args):
+    try:
+        result = run(*args)
+        result_queue.put(("success", result))
+    except Exception as e:
+        result_queue.put(("error", str(e)))
+
+
 def main():
     storage_backends = "."
     device_id = 1
@@ -99,25 +108,40 @@ def main():
                         io_num = int(num_tokens / block_len * block_layer)
 
                         # Run test and get results
+                        result_queue = multiprocessing.Queue()
+
+                        process = multiprocessing.Process(
+                            target=run_wrapper,
+                            args=(
+                                result_queue,
+                                storage_backends,
+                                device_id,
+                                repeat,
+                                num_head,
+                                block_len,
+                                transferStreamNumber,
+                                num_tokens,
+                                block_layer,
+                                head_size,
+                                block_elem_size,
+                            ),
+                        )
+
+                        process.start()
+                        process.join()
+
+                        status, result = result_queue.get()
+                        if status == "error":
+                            raise Exception(f"Error in subprocess: {result}")
+
                         (
                             avg_w_size,
                             avg_w_time,
                             avg_w_bw,
                             avg_r_time,
                             avg_r_bw,
                             avg_r_size,
-                        ) = run(
-                            storage_backends,
-                            device_id,
-                            repeat,
-                            num_head,
-                            block_len,
-                            transferStreamNumber,
-                            num_tokens,
-                            block_layer,
-                            head_size,
-                            block_elem_size,
-                        )
+                        ) = result
 
                         writer.writerow(
                             [