diff --git a/jenkins/scripts/slurm_run.sh b/jenkins/scripts/slurm_run.sh index 717f1be791a..ce2712881b7 100755 --- a/jenkins/scripts/slurm_run.sh +++ b/jenkins/scripts/slurm_run.sh @@ -61,6 +61,11 @@ else done fi +if [[ "$stageName" == *GB200* ]]; then + echo "Checking Coherent GPU mapping (for GB200)..." + grep Coherent /proc/driver/nvidia/params || echo "Unable to grep Coherent from /proc/driver/nvidia/params" +fi + llmapiLaunchScript="$llmSrcNode/tensorrt_llm/llmapi/trtllm-llmapi-launch" chmod +x $llmapiLaunchScript cd $llmSrcNode/tests/integration/defs