Merge remote-tracking branch 'upstream/main'

Author: dchourasia

tests/common/environment.go

@@ -42,18 +42,20 @@ const (
 )
 
 const (
-	tierSmoke   = "Smoke"
-	tierSanity  = "Sanity"
-	tier1       = "Tier1"
-	tier2       = "Tier2"
-	tier3       = "Tier3"
-	preUpgrade  = "Pre-Upgrade"
-	postUpgrade = "Post-Upgrade"
-	kftoCuda    = "KFTO-CUDA"
-	kftoRocm    = "KFTO-ROCm"
+	tierSmoke    = "Smoke"
+	tierSanity   = "Sanity"
+	tier1        = "Tier1"
+	tier2        = "Tier2"
+	tier3        = "Tier3"
+	preUpgrade   = "Pre-Upgrade"
+	postUpgrade  = "Post-Upgrade"
+	kftoCuda     = "KFTO-CUDA"
+	kftoRocm     = "KFTO-ROCm"
+	examplesCuda = "Examples-CUDA"
+	examplesRocm = "Examples-ROCm"
 )
 
-var testTiers = []string{tierSmoke, tierSanity, tier1, tier2, tier3, preUpgrade, postUpgrade, kftoCuda, kftoRocm}
+var testTiers = []string{tierSmoke, tierSanity, tier1, tier2, tier3, preUpgrade, postUpgrade, kftoCuda, kftoRocm, examplesCuda, examplesRocm}
 
 var testTierParam string
 

tests/common/test_tag.go

@@ -77,6 +77,14 @@ var KftoRocm = func(test Test) (runTest bool, skipReason string) {
 	return testTier(test, kftoRocm)
 }
 
+var ExamplesCuda = func(test Test) (runTest bool, skipReason string) {
+	return testTier(test, examplesCuda)
+}
+
+var ExamplesRocm = func(test Test) (runTest bool, skipReason string) {
+	return testTier(test, examplesRocm)
+}
+
 func Gpu(accelerator Accelerator) func(test Test) (runTest bool, skipReason string) {
 	return func(test Test) (runTest bool, skipReason string) {
 		return isGpuCountAvailableForNodes(test, 1, accelerator.ResourceLabel, 1)

tests/kfto/kfto_sft_llm_test.go

@@ -39,12 +39,12 @@ import (
 )
 
 func TestKftoSftLlmLlama3_1_8BInstructWithCudaPyTorch251(t *testing.T) {
-	Tags(t, KftoCuda)
+	Tags(t, ExamplesCuda)
 	kftoSftLlm(t, GetTrainingCudaPyTorch251Image(), NVIDIA, "meta-llama/Llama-3.1-8B-Instruct")
 }
 
 func TestKftoSftLlmLlama3_1_8BInstructWithROCmPyTorch251(t *testing.T) {
-	Tags(t, KftoRocm)
+	Tags(t, ExamplesRocm)
 	kftoSftLlm(t, GetTrainingROCmPyTorch251Image(), AMD, "meta-llama/Llama-3.1-8B-Instruct")
 }