Add debug info

wbo4958 · Oct 16, 2024 · ac622ed · ac622ed
1 parent 3f9bfaf
commit ac622ed
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/python-package/xgboost/spark/core.py b/python-package/xgboost/spark/core.py
@@ -1026,6 +1026,11 @@ def _train_booster(
             from pyspark import BarrierTaskContext
 
             context = BarrierTaskContext.get()
+            get_logger(_LOG_TAG, log_level).info("before entering barrier.")
+            print("before entering barrier.")
+            context.barrier()
+            get_logger(_LOG_TAG, log_level).info("after entering barrier.")
+            print("after entering barrier.")
 
             dev_ordinal = None
             use_qdm = _can_use_qdm(