@@ -67,16 +67,12 @@ var _ = Describe("GPU instance support", func() {
6767 gpuInstanceType : "g5.12xlarge" ,
6868 amiFamily : api .NodeImageFamilyAmazonLinux2 ,
6969 }),
70- Entry ("Ubuntu2004" , gpuInstanceEntry {
71- amiFamily : api .NodeImageFamilyUbuntu2004 ,
72- gpuInstanceType : "g4dn.xlarge" ,
73- }),
7470 Entry ("Ubuntu2204" , gpuInstanceEntry {
75- amiFamily : api .NodeImageFamilyUbuntu2004 ,
71+ amiFamily : api .NodeImageFamilyUbuntu2204 ,
7672 gpuInstanceType : "g4dn.xlarge" ,
7773 }),
7874 Entry ("UbuntuPro2204" , gpuInstanceEntry {
79- amiFamily : api .NodeImageFamilyUbuntu2004 ,
75+ amiFamily : api .NodeImageFamilyUbuntuPro2204 ,
8076 gpuInstanceType : "g4dn.xlarge" ,
8177 }),
8278 Entry ("Bottlerocket INF" , gpuInstanceEntry {
@@ -164,8 +160,8 @@ var _ = Describe("GPU instance support", func() {
164160 amiFamily : api .NodeImageFamilyBottlerocket ,
165161 gpuInstanceType : "g4dn.xlarge" ,
166162 }),
167- Entry ("Ubuntu2004 " , gpuInstanceEntry {
168- amiFamily : api .NodeImageFamilyUbuntu2004 ,
163+ Entry ("UbuntuPro2004 " , gpuInstanceEntry {
164+ amiFamily : api .NodeImageFamilyUbuntuPro2004 ,
169165 gpuInstanceType : "g4dn.xlarge" ,
170166 }),
171167 Entry ("Windows2019Core" , gpuInstanceEntry {
@@ -218,13 +214,6 @@ var _ = Describe("GPU instance support", func() {
218214 Expect (output .String ()).NotTo (ContainSubstring (api .GPUDriversWarning (mng .AMIFamily )))
219215 }
220216 },
221- Entry ("Windows without GPU instances" , gpuInstanceEntry {
222- amiFamily : api .NodeImageFamilyUbuntu2004 ,
223- instanceSelector : & api.InstanceSelector {
224- VCPUs : 4 ,
225- GPUs : newInt (0 ),
226- },
227- }),
228217 Entry ("Windows with explicit GPU instance" , gpuInstanceEntry {
229218 amiFamily : api .NodeImageFamilyWindowsServer2019FullContainer ,
230219 gpuInstanceType : "g4dn.xlarge" ,
@@ -237,14 +226,6 @@ var _ = Describe("GPU instance support", func() {
237226 },
238227 expectWarning : true ,
239228 }),
240- Entry ("Ubuntu with implicit GPU instance" , gpuInstanceEntry {
241- amiFamily : api .NodeImageFamilyUbuntu2004 ,
242- instanceSelector : & api.InstanceSelector {
243- VCPUs : 4 ,
244- GPUs : newInt (2 ),
245- },
246- expectWarning : true ,
247- }),
248229 )
249230
250231 Describe ("No GPU instance type support validation for custom AMI" , func () {
@@ -272,7 +253,6 @@ var _ = Describe("GPU instance support", func() {
272253 },
273254 Entry ("AmazonLinux2" , api .NodeImageFamilyAmazonLinux2 , true ),
274255 Entry ("AmazonLinux2023" , api .NodeImageFamilyAmazonLinux2023 , false ),
275- Entry ("Ubuntu2004" , api .NodeImageFamilyUbuntu2004 , false ),
276256 Entry ("Windows2019Full" , api .NodeImageFamilyWindowsServer2019FullContainer , true ),
277257 Entry ("Windows2019Core" , api .NodeImageFamilyWindowsServer2019CoreContainer , true ),
278258 Entry ("Bottlerocket" , api .NodeImageFamilyBottlerocket , false ),
0 commit comments