Skip to content

Commit 8ad8367

Browse files
add gpu resource to ClusterQueue
1 parent 6dbb295 commit 8ad8367

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

tests/kfto/core/kfto_kueue_sft_test.go

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -141,7 +141,7 @@ func TestPytorchjobUsingKueueQuota(t *testing.T) {
141141
NamespaceSelector: &metav1.LabelSelector{},
142142
ResourceGroups: []kueuev1beta1.ResourceGroup{
143143
{
144-
CoveredResources: []corev1.ResourceName{corev1.ResourceName("cpu"), corev1.ResourceName("memory")},
144+
CoveredResources: []corev1.ResourceName{corev1.ResourceName("cpu"), corev1.ResourceName("memory"), corev1.ResourceName("nvidia.com/gpu")},
145145
Flavors: []kueuev1beta1.FlavorQuotas{
146146
{
147147
Name: kueuev1beta1.ResourceFlavorReference(resourceFlavor.Name),
@@ -154,6 +154,10 @@ func TestPytorchjobUsingKueueQuota(t *testing.T) {
154154
Name: corev1.ResourceMemory,
155155
NominalQuota: resource.MustParse("8Gi"),
156156
},
157+
{
158+
Name: corev1.ResourceName("nvidia.com/gpu"),
159+
NominalQuota: resource.MustParse("0"),
160+
},
157161
},
158162
},
159163
},

0 commit comments

Comments
 (0)