moved gsutil copy to condtional to avoid breakages

MaxText/profiler.py

@@ -18,6 +18,7 @@
 from ctypes import cdll
 import os
 import subprocess
+import shutil
 
 import max_logging
 
@@ -61,6 +62,9 @@ def deactivate(self):
  max_logging.log("WARNING: library for nsys was not loaded \n" "profiler has no effect")
  return
  # Popen() instead of run() for non-blocking behavior
- subprocess.Popen(["gsutil", "cp", "*nsys-rep", self.output_path]) # pylint: disable=consider-using-with
+ if shutil.which('gsutil') is not None:
+ subprocess.Popen(["gsutil", "cp", "*nsys-rep", self.output_path]) # pylint: disable=consider-using-with
+ else:
+ max_logging.log("WARNING: gsutil is not installed or not found in the system's PATH. Skipping upload...")
  elif self.mode == "xplane":
  jax.profiler.stop_trace()