diff --git a/Ironwood/src/benchmark_host_device.py b/Ironwood/src/benchmark_host_device.py index cc1e505..6d47dab 100644 --- a/Ironwood/src/benchmark_host_device.py +++ b/Ironwood/src/benchmark_host_device.py @@ -96,8 +96,8 @@ def benchmark_host_device( device_array.delete() return { - "H2D_Bandwidth_ms": h2d_perf, - "D2H_Bandwidth_ms": d2h_perf, + "h2d_bandwidth_ms": h2d_perf, + "d2h_bandwidth_ms": d2h_perf, } diff --git a/Ironwood/src/benchmark_utils.py b/Ironwood/src/benchmark_utils.py index 9e0baa4..0136012 100644 --- a/Ironwood/src/benchmark_utils.py +++ b/Ironwood/src/benchmark_utils.py @@ -804,7 +804,14 @@ def upload_to_storage(trace_dir: str, local_file: str): if trace_dir.startswith("gs://"): # Google Cloud Storage (GCS) try: subprocess.run( - ["gcloud", "storage", "cp", "--recursive", local_file, trace_dir], + [ + "gcloud", + "storage", + "cp", + "--recursive", + local_file, + trace_dir, + ], check=True, capture_output=True, ) diff --git a/src/benchmark_utils.py b/src/benchmark_utils.py index a4ef714..340836b 100644 --- a/src/benchmark_utils.py +++ b/src/benchmark_utils.py @@ -243,7 +243,14 @@ def upload_to_storage(trace_dir: str, local_file: str): if trace_dir.startswith("gs://"): # Google Cloud Storage (GCS) try: subprocess.run( - ["gcloud", "storage", "cp", "--recursive", local_file, trace_dir], + [ + "gcloud", + "storage", + "cp", + "--recursive", + local_file, + trace_dir, + ], check=True, capture_output=True, )