@@ -182,14 +182,41 @@ def is_nvidia_superchip(gpu_name: str) -> bool:
182182 NvidiaGPUInfo (name = "L40" , memory = 48 , compute_capability = (8 , 9 )),
183183 NvidiaGPUInfo (name = "L40S" , memory = 48 , compute_capability = (8 , 9 )),
184184 NvidiaGPUInfo (name = "P100" , memory = 16 , compute_capability = (6 , 0 )),
185+ NvidiaGPUInfo (name = "RTX2060" , memory = 6 , compute_capability = (7 , 5 )),
186+ NvidiaGPUInfo (name = "RTX2060" , memory = 12 , compute_capability = (7 , 5 )),
187+ NvidiaGPUInfo (name = "RTX2060SUPER" , memory = 8 , compute_capability = (7 , 5 )),
188+ NvidiaGPUInfo (name = "RTX2070" , memory = 8 , compute_capability = (7 , 5 )),
189+ NvidiaGPUInfo (name = "RTX2070SUPER" , memory = 8 , compute_capability = (7 , 5 )),
190+ NvidiaGPUInfo (name = "RTX2080" , memory = 8 , compute_capability = (7 , 5 )),
191+ NvidiaGPUInfo (name = "RTX2080SUPER" , memory = 8 , compute_capability = (7 , 5 )),
192+ NvidiaGPUInfo (name = "RTX2080Ti" , memory = 11 , compute_capability = (7 , 5 )),
193+ NvidiaGPUInfo (name = "RTX3050" , memory = 6 , compute_capability = (8 , 6 )),
194+ NvidiaGPUInfo (name = "RTX3050" , memory = 8 , compute_capability = (8 , 6 )),
185195 NvidiaGPUInfo (name = "RTX3060" , memory = 8 , compute_capability = (8 , 6 )),
186196 NvidiaGPUInfo (name = "RTX3060" , memory = 12 , compute_capability = (8 , 6 )),
187197 NvidiaGPUInfo (name = "RTX3060Ti" , memory = 8 , compute_capability = (8 , 6 )),
198+ NvidiaGPUInfo (name = "RTX3070" , memory = 8 , compute_capability = (8 , 6 )),
188199 NvidiaGPUInfo (name = "RTX3070Ti" , memory = 8 , compute_capability = (8 , 6 )),
189200 NvidiaGPUInfo (name = "RTX3080" , memory = 10 , compute_capability = (8 , 6 )),
190201 NvidiaGPUInfo (name = "RTX3080Ti" , memory = 12 , compute_capability = (8 , 6 )),
191202 NvidiaGPUInfo (name = "RTX3090" , memory = 24 , compute_capability = (8 , 6 )),
203+ NvidiaGPUInfo (name = "RTX3090Ti" , memory = 24 , compute_capability = (8 , 6 )),
204+ NvidiaGPUInfo (name = "RTX4060" , memory = 8 , compute_capability = (8 , 9 )),
205+ NvidiaGPUInfo (name = "RTX4060Ti" , memory = 8 , compute_capability = (8 , 9 )),
206+ NvidiaGPUInfo (name = "RTX4060Ti" , memory = 16 , compute_capability = (8 , 9 )),
207+ NvidiaGPUInfo (name = "RTX4070" , memory = 12 , compute_capability = (8 , 9 )),
208+ NvidiaGPUInfo (name = "RTX4070SUPER" , memory = 12 , compute_capability = (8 , 9 )),
209+ NvidiaGPUInfo (name = "RTX4070Ti" , memory = 12 , compute_capability = (8 , 9 )),
210+ NvidiaGPUInfo (name = "RTX4070TiSUPER" , memory = 16 , compute_capability = (8 , 9 )),
211+ NvidiaGPUInfo (name = "RTX4080" , memory = 16 , compute_capability = (8 , 9 )),
212+ NvidiaGPUInfo (name = "RTX4080SUPER" , memory = 16 , compute_capability = (8 , 9 )),
192213 NvidiaGPUInfo (name = "RTX4090" , memory = 24 , compute_capability = (8 , 9 )),
214+ NvidiaGPUInfo (name = "RTX5060" , memory = 8 , compute_capability = (12 , 0 )),
215+ NvidiaGPUInfo (name = "RTX5060Ti" , memory = 8 , compute_capability = (12 , 0 )),
216+ NvidiaGPUInfo (name = "RTX5060Ti" , memory = 16 , compute_capability = (12 , 0 )),
217+ NvidiaGPUInfo (name = "RTX5070" , memory = 12 , compute_capability = (12 , 0 )),
218+ NvidiaGPUInfo (name = "RTX5070Ti" , memory = 16 , compute_capability = (12 , 0 )),
219+ NvidiaGPUInfo (name = "RTX5080" , memory = 16 , compute_capability = (12 , 0 )),
193220 NvidiaGPUInfo (name = "RTX5090" , memory = 32 , compute_capability = (12 , 0 )),
194221 NvidiaGPUInfo (name = "RTX6000" , memory = 24 , compute_capability = (7 , 5 )),
195222 NvidiaGPUInfo (name = "RTX2000Ada" , memory = 16 , compute_capability = (8 , 9 )),
0 commit comments