Allow set XLA_FLAGS with --debug_dump_gcs

Author: PatriosTheGreat

src/xpk/core/core.py

@@ -233,13 +233,17 @@ def parse_env_config(args, tensorboard_config, system: SystemCharacteristics):
 
   if not args.use_pathways:
     if args.debug_dump_gcs:
-      if 'XLA_FLAGS' in env:
+      if 'XLA_FLAGS' in env and '--xla_dump_to' in env['XLA_FLAGS']:
         raise ValueError(
-            'Conflict: XLA_FLAGS defined in both --debug_dump_gcs '
-            'and environment file. Please choose one way to define '
-            'XLA_FLAGS.'
+            'Conflict: --xla_dump_to defined in both --debug_dump_gcs '
+            'and environment file  via XLA_FLAGS. Please choose one way '
+            'to define it.'
         )
-      env['XLA_FLAGS'] = '--xla_dump_to=/tmp/xla_dump/'
+      other_flags = ''
+      if 'XLA_FLAGS' in env:
+        other_flags = env['XLA_FLAGS']
+
+      env['XLA_FLAGS'] = '--xla_dump_to=/tmp/xla_dump/ ' + other_flags
 
     if tensorboard_config:
       env['UPLOAD_DATA_TO_TENSORBOARD'] = True