debug prints

Author: apbose

.github/scripts/filter-matrix.py

@@ -150,6 +150,8 @@ def main(args: list[str]) -> None:
             options.limit_pr_builds == "true",
         ):
             print(f"[DEBUG] passed filter - adding to build matrix", file=sys.stderr)
+            # Add tensorrt version to all items (required by linux-test.yml)
+            item["tensorrt"] = {"version": "10.13.3"}
             filtered_includes.append(item)
 
             # NEW: Create distributed variant for specific configs

.github/workflows/build-test-linux-x86_64.yml

@@ -519,7 +519,7 @@ jobs:
 
   L2-dynamo-distributed-tests:
     name: L2 dynamo distributed tests
-    needs: [filter-matrix, build, L1-dynamo-core-tests, L1-dynamo-compile-tests, L1-torch-compile-tests, L1-torchscript-tests]
+    needs: [filter-matrix, build]
     strategy:
       fail-fast: false
       matrix:
@@ -542,6 +542,8 @@ jobs:
           "include": ${{ toJSON(fromJSON(needs.filter-matrix.outputs.matrix).distributed_include) }}
         }
       pre-script: ${{ matrix.pre-script }}
+      post-script: ${{ matrix.post-script }}
+      smoke-test-script: ${{ matrix.smoke-test-script }}
       script: |
         set -euo pipefail
 
@@ -551,9 +553,7 @@ jobs:
         echo "=========================================="
         echo "Python version: ${PYTHON_VERSION}"
         echo "CUDA version: ${CU_VERSION}"
-        echo "Runner: ${{ matrix.validation_runner }}"
-        echo "Num GPUs: ${{ matrix.num_gpus }}"
-        echo "Config: ${{ matrix.config }}"
+        echo "Num GPUs: ${NUM_GPUS}"
         echo "=========================================="
 
         # Verify GPUs are available
@@ -568,17 +568,36 @@ jobs:
 
         # Install MPI (required for TensorRT-LLM plugins)
         echo "Installing MPI..."
-        dnf install -y mpich mpich-devel openmpi openmpi-devel
+        dnf install -y openmpi openmpi-devel
+
+        # Add OpenMPI to PATH (RHEL/AlmaLinux specific location)
+        export PATH="/usr/lib64/openmpi/bin:$PATH"
+        export LD_LIBRARY_PATH="/usr/lib64/openmpi/lib:$LD_LIBRARY_PATH"
+
+        # Verify mpirun is accessible
+        which mpirun
+        mpirun --version
 
         # Run distributed tests
         pushd .
         cd tests/py/dynamo
 
         echo "Running distributed tests with mpirun..."
-        mpirun --allow-run-as-root -n ${{ matrix.num_gpus }} \
-          python -m pytest -ra \
-          --junitxml=${RUNNER_TEST_RESULTS_DIR}/l2_dynamo_distributed_test_results.xml \
-          distributed/test_nccl_ops.py
+        echo "[CONFIG] Number of GPUs to use: ${NUM_GPUS}"
+        echo "[AVAILABLE] GPUs detected by nvidia-smi: $(nvidia-smi --query-gpu=name --format=csv,noheader | wc -l)"
+
+        # Use a wrapper script to ensure only rank 0 writes the JUnit XML
+        # Each rank runs pytest, but only rank 0 saves results to avoid file conflicts
+        RANK_0_XML="${RUNNER_TEST_RESULTS_DIR}/l2_dynamo_distributed_test_results.xml"
+        mpirun --allow-run-as-root -n ${NUM_GPUS} \
+          bash -c '
+            echo "[MPI DEBUG] Rank: ${OMPI_COMM_WORLD_RANK:-0}, World Size: ${OMPI_COMM_WORLD_SIZE:-1}"
+            if [ "${OMPI_COMM_WORLD_RANK:-0}" -eq 0 ]; then
+              python -m pytest -ra --junitxml='"${RANK_0_XML}"' distributed/test_nccl_ops.py
+            else
+              python -m pytest -ra distributed/test_nccl_ops.py
+            fi
+          '
 
         popd
 

.github/workflows/linux-test.yml

@@ -74,6 +74,7 @@ jobs:
       RUNNER_TEST_RESULTS_DIR: /tmp/test_results
       ARCH: ${{ inputs.architecture }}
       USE_TRT_RTX: ${{ inputs.use-rtx }}
+      NUM_GPUS: ${{ matrix.num_gpus || '' }}
       DOWNLOAD_ARTIFACT_NAME: pytorch_tensorrt_${{ matrix.tensorrt.version }}_${{ matrix.python_version }}_${{ matrix.desired_cuda }}_${{ inputs.architecture }}
     name: ${{ inputs.job-name }}-${{ matrix.tensorrt.version }}-${{ matrix.python_version }}-${{ matrix.desired_cuda }}
     runs-on: ${{ matrix.validation_runner }}