add gpu resource to ClusterQueue

opendatahub-io · Nov 22, 2024 · 8ad8367 · 8ad8367
1 parent 6dbb295
commit 8ad8367
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/kfto/core/kfto_kueue_sft_test.go b/tests/kfto/core/kfto_kueue_sft_test.go
@@ -141,7 +141,7 @@ func TestPytorchjobUsingKueueQuota(t *testing.T) {
 		NamespaceSelector: &metav1.LabelSelector{},
 		ResourceGroups: []kueuev1beta1.ResourceGroup{
 			{
-				CoveredResources: []corev1.ResourceName{corev1.ResourceName("cpu"), corev1.ResourceName("memory")},
+				CoveredResources: []corev1.ResourceName{corev1.ResourceName("cpu"), corev1.ResourceName("memory"), corev1.ResourceName("nvidia.com/gpu")},
 				Flavors: []kueuev1beta1.FlavorQuotas{
 					{
 						Name: kueuev1beta1.ResourceFlavorReference(resourceFlavor.Name),
@@ -154,6 +154,10 @@ func TestPytorchjobUsingKueueQuota(t *testing.T) {
 								Name:         corev1.ResourceMemory,
 								NominalQuota: resource.MustParse("8Gi"),
 							},
+							{
+								Name:         corev1.ResourceName("nvidia.com/gpu"),
+								NominalQuota: resource.MustParse("0"),
+							},
 						},
 					},
 				},