Skip to content

Commit

Permalink
Intel/CI: Update oneCCL GPU stage to use torchic partition
Browse files Browse the repository at this point in the history
OneCCL GPU tests require to use new node which is available with
torchic partition.

Signed-off-by: Juee Himalbhai Desai <[email protected]>
  • Loading branch information
Juee14Desai authored and zachdworkin committed May 30, 2024
1 parent 7db99ae commit 3967943
Showing 1 changed file with 8 additions and 5 deletions.
13 changes: 8 additions & 5 deletions contrib/intel/jenkins/Jenkinsfile
Original file line number Diff line number Diff line change
Expand Up @@ -763,15 +763,18 @@ pipeline {
steps {
script {
dir (RUN_LOCATION) {
run_middleware([["tcp", null]], "oneCCL-GPU-v3", "onecclgpu",
"gpu", "fabrics-ci", "2", null, null,
"FI_HMEM_DISABLE_P2P=1")
run_middleware([["psm3", null]], "oneCCL-GPU-v3", "onecclgpu",
"gpu", "fabrics-ci", "2", null, null,
"gpu", "torchic", "1", null, null,
"FI_HMEM_DISABLE_P2P=1")
run_middleware([["verbs", null]], "oneCCL-GPU-v3", "onecclgpu",
"gpu", "fabrics-ci", "2", null, null,
"gpu", "torchic", "1", null, null,
"FI_HMEM_DISABLE_P2P=1")
run_middleware([["tcp", null]], "oneCCL-GPU-v3", "onecclgpu",
"gpu", "torchic", "1", null, null,
"FI_HMEM_DISABLE_P2P=1")
run_middleware([["shm", null]], "oneCCL-GPU-v3", "onecclgpu",
"gpu", "torchic", "1", null, null,
"FI_HMEM_DISABLE_P2P=1")
}
}
}
Expand Down

0 comments on commit 3967943

Please sign in to comment.