Skip to content

Commit d54d39f

Browse files
Update hardware.ts (#718)
add some unique gpus --------- Co-authored-by: Pedro Cuenca <[email protected]>
1 parent d39262c commit d54d39f

File tree

1 file changed

+39
-7
lines changed

1 file changed

+39
-7
lines changed

packages/tasks/src/hardware.ts

Lines changed: 39 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -84,14 +84,14 @@ export const SKUS = {
8484
tflops: 31.24,
8585
memory: [24],
8686
},
87-
T4: {
88-
tflops: 65.13,
89-
memory: [16],
90-
},
9187
"RTX 4090": {
9288
tflops: 82.58,
9389
memory: [24],
9490
},
91+
"RTX 4090D": {
92+
tflops: 79.49,
93+
memory: [24],
94+
},
9595
"RTX 4080 SUPER": {
9696
tflops: 52.2,
9797
memory: [16],
@@ -172,17 +172,37 @@ export const SKUS = {
172172
tflops: 9.22,
173173
memory: [6],
174174
},
175+
"GTX 1080 Ti": {
176+
tflops: 11.34, // float32 (GPU does not support native float16)
177+
memory: [11],
178+
},
179+
"GTX 1070 Ti": {
180+
tflops: 8.2, // float32 (GPU does not support native float16)
181+
memory: [8],
182+
},
175183
"RTX Titan": {
176184
tflops: 32.62,
177185
memory: [24],
178186
},
187+
"GTX 1650 Mobile": {
188+
tflops: 6.39,
189+
memory: [4],
190+
},
191+
T4: {
192+
tflops: 65.13,
193+
memory: [16],
194+
},
179195
V100: {
180196
tflops: 28.26,
181197
memory: [32, 16],
182198
},
183-
"GTX 1650 Mobile": {
184-
tflops: 6.39,
185-
memory: [4],
199+
"Quadro P6000": {
200+
tflops: 12.63, // float32 (GPU does not support native float16)
201+
memory: [24],
202+
},
203+
P40: {
204+
tflops: 11.76, // float32 (GPU does not support native float16)
205+
memory: [24],
186206
},
187207
},
188208
AMD: {
@@ -226,6 +246,18 @@ export const SKUS = {
226246
tflops: 45.14,
227247
memory: [16, 8],
228248
},
249+
"RX 6950 XT": {
250+
tflops: 47.31,
251+
memory: [16],
252+
},
253+
"RX 6800": {
254+
tflops: 32.33,
255+
memory: [16],
256+
},
257+
"Radeon Pro VII": {
258+
tflops: 26.11,
259+
memory: [16],
260+
},
229261
},
230262
},
231263
CPU: {

0 commit comments

Comments
 (0)