add some prints

QITI · Feb 27, 2024 · dc71bb0 · dc71bb0
1 parent c38eed8
commit dc71bb0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/runtime_benchmark/runtime_ifta.py b/runtime_benchmark/runtime_ifta.py
@@ -56,6 +56,7 @@ def task(method):
     print("Is Built with CUDA: ", tf.test.is_built_with_cuda())
     print("CUDA Version: ", tf.sysconfig.get_build_info()["cuda_version"])
     # Run the task on GPU
+    print("Running on GPU")
     start_time = time.time()
     task('ifta')
     gpu_time = time.time() - start_time

diff --git a/runtime_benchmark/slm_hologram_gs.py b/runtime_benchmark/slm_hologram_gs.py
@@ -30,7 +30,9 @@ def task(method):
 # Disable GPU for the first run
 os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
 print("Num GPUs Available: ", len(tf.config.list_physical_devices('GPU')))
+
 # Run the task on CPU
+print("Running on CPU")
 start_time = time.time()
 task('gs')
 cpu_time = time.time() - start_time