@@ -35,11 +35,12 @@ type ClusterTrainingRuntime struct {
3535}
3636
3737var expectedRuntimes = []ClusterTrainingRuntime {
38- {Name : "torch-distributed" , RHOAIImage : "odh-training-cuda128-torch28-py312-rhel9" },
39- {Name : "torch-distributed-rocm" , RHOAIImage : "odh-training-rocm64-torch28-py312-rhel9" },
40- {Name : "torch-distributed-th03-cuda128-torch28-py312" , RHOAIImage : "odh-training-cuda128-torch28-py312-rhel9" },
41- {Name : "training-hub" , RHOAIImage : "odh-training-cuda128-torch28-py312-rhel9" },
42- {Name : "training-hub03-cuda128-torch28-py312" , RHOAIImage : "odh-training-cuda128-torch28-py312-rhel9" },
38+ {Name : "torch-distributed" , RHOAIImage : "odh-training-cuda128-torch29-py312-rhel9" },
39+ {Name : "torch-distributed-rocm" , RHOAIImage : "odh-training-rocm64-torch29-py312-rhel9" },
40+ {Name : "torch-distributed-cuda128-torch29-py312" , RHOAIImage : "odh-training-cuda128-torch29-py312-rhel9" },
41+ {Name : "torch-distributed-rocm64-torch29-py312" , RHOAIImage : "odh-training-rocm64-torch29-py312-rhel9" },
42+ {Name : "training-hub" , RHOAIImage : "odh-training-cuda128-torch29-py312-rhel9" },
43+ {Name : "training-hub-th05-cuda128-torch29-py312" , RHOAIImage : "odh-training-cuda128-torch29-py312-rhel9" },
4344}
4445
4546// defaultClusterTrainingRuntime is used across integration tests
0 commit comments