Insert needed space as seperator.

trentlo · trentlo · commit 388f1acbc076 · 2020-07-22T19:08:55.000-07:00
diff --git a/TensorFlow/LanguageModeling/BERT/run_squad.py b/TensorFlow/LanguageModeling/BERT/run_squad.py
@@ -936,12 +936,12 @@ def serving_input_fn():
 def main(_):
   # causes memory fragmentation for bert leading to OOM
   if os.environ.get("TF_XLA_FLAGS", None) is not None:
-    os.environ["TF_XLA_FLAGS"] += "--tf_xla_enable_lazy_compilation=false"
+    os.environ["TF_XLA_FLAGS"] += " --tf_xla_enable_lazy_compilation=false"
   else:
     os.environ["TF_XLA_FLAGS"] = "--tf_xla_enable_lazy_compilation=false"
 
   # Enable async_io to speed up multi-gpu training with XLA and Horovod.
-  os.environ["TF_XLA_FLAGS"] += "--tf_xla_async_io_level=1"
+  os.environ["TF_XLA_FLAGS"] += " --tf_xla_async_io_level=1"
 
   tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.INFO)
   dllogging = utils.dllogger_class.dllogger_class(FLAGS.dllog_path)
