Skip to content

Commit 2894ab1

Browse files
ChughShilpasutaakar
authored andcommitted
Update trainer tests with new ClusterTraining Runtimes
1 parent 867d932 commit 2894ab1

1 file changed

Lines changed: 6 additions & 5 deletions

File tree

tests/trainer/cluster_training_runtimes_test.go

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -35,11 +35,12 @@ type ClusterTrainingRuntime struct {
3535
}
3636

3737
var 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

Comments
 (0)