-
Notifications
You must be signed in to change notification settings - Fork 242
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
I hesitated putting this one in `@huggingface/tasks`, or creating a new `@huggingface/hardware`. What do you think? The idea is the community to be able to contribute to that list. ### What I picked for now⤵️ Because I'm lazy – and because it's somewhat linked to #659 – i've added it to `tasks`. --------- Co-authored-by: Pedro Cuenca <[email protected]> Co-authored-by: apolinário <[email protected]>
- Loading branch information
1 parent
674f945
commit bba3138
Showing
2 changed files
with
311 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,309 @@ | ||
/** | ||
* Biden AI Executive Order | ||
* https://www.whitehouse.gov/briefing-room/presidential-actions/2023/10/30/executive-order-on-the-safe-secure-and-trustworthy-development-and-use-of-artificial-intelligence/ | ||
*/ | ||
export const TFLOPS_THRESHOLD_WHITE_HOUSE_MODEL_TRAINING_TOTAL = 10 ** 14; | ||
export const TFLOPS_THRESHOLD_WHITE_HOUSE_MODEL_TRAINING_TOTAL_BIOLOGY = 10 ** 11; | ||
export const TFLOPS_THRESHOLD_WHITE_HOUSE_CLUSTER = 10 ** 8; | ||
|
||
/** | ||
* EU AI Act | ||
* https://ec.europa.eu/commission/presscorner/detail/en/qanda_21_1683 | ||
*/ | ||
export const TFLOPS_THRESHOLD_EU_AI_ACT_MODEL_TRAINING_TOTAL = 10 ** 13; | ||
|
||
export interface HardwareSpec { | ||
/** | ||
* Approximate value, in FP16 whenever possible. | ||
* This is only approximate/theoretical and shouldn't be taken too seriously. | ||
* Currently the CPU values are from cpu-monkey.com | ||
* while the GPU values are from techpowerup.com | ||
* | ||
* Note to reviewers: I got fed up with data entry, | ||
* and HuggingChat running Llama3 with Web search was failing a bit, | ||
* so some of those values might be slightly inaccurate. Forgive me and please feel free to improve. | ||
*/ | ||
tflops: number; | ||
/** | ||
* If an array is specified, options of memory size (can be VRAM, unified RAM) | ||
* e.g. an A100 exists in 40 or 80 GB. | ||
*/ | ||
memory?: number[]; | ||
} | ||
|
||
export const DEFAULT_MEMORY_OPTIONS = [8, 16, 24, 32, 40, 48, 64, 80, 96, 128]; | ||
|
||
export const SKUS = { | ||
GPU: { | ||
NVIDIA: { | ||
H100: { | ||
tflops: 267.6, | ||
memory: [80], | ||
}, | ||
L40: { | ||
tflops: 90.52, | ||
memory: [48], | ||
}, | ||
"RTX 6000 Ada": { | ||
tflops: 91.1, | ||
memory: [48], | ||
}, | ||
A100: { | ||
tflops: 77.97, | ||
memory: [80, 40], | ||
}, | ||
A40: { | ||
tflops: 37.42, | ||
memory: [48], | ||
}, | ||
A10: { | ||
tflops: 31.24, | ||
memory: [24], | ||
}, | ||
T4: { | ||
tflops: 65.13, | ||
memory: [16], | ||
}, | ||
"RTX 4090": { | ||
tflops: 82.58, | ||
memory: [24], | ||
}, | ||
"RTX 4080 SUPER": { | ||
tflops: 52.2, | ||
memory: [16], | ||
}, | ||
"RTX 4080": { | ||
tflops: 48.7, | ||
memory: [16], | ||
}, | ||
"RTX 4070": { | ||
tflops: 29.15, | ||
memory: [12], | ||
}, | ||
"RTX 4070 Ti": { | ||
tflops: 40.09, | ||
memory: [12], | ||
}, | ||
"RTX 4070 Super": { | ||
tflops: 35.48, | ||
memory: [12], | ||
}, | ||
"RTX 4070 Ti Super": { | ||
tflops: 44.1, | ||
memory: [16], | ||
}, | ||
"RTX 3090": { | ||
tflops: 35.58, | ||
memory: [24], | ||
}, | ||
"RTX 3090 Ti": { | ||
tflops: 40, | ||
memory: [24], | ||
}, | ||
"RTX 3080 Ti": { | ||
tflops: 34.1, | ||
memory: [12], | ||
}, | ||
"RTX 3080": { | ||
tflops: 30.6, | ||
memory: [12, 10], | ||
}, | ||
}, | ||
AMD: { | ||
MI300: { | ||
tflops: 383.0, | ||
memory: [192], | ||
}, | ||
MI250: { | ||
tflops: 362.1, | ||
memory: [128], | ||
}, | ||
MI210: { | ||
tflops: 181.0, | ||
memory: [64], | ||
}, | ||
"RX 7900 XTX": { | ||
tflops: 122.8, | ||
memory: [24], | ||
}, | ||
"RX 7900 XT": { | ||
tflops: 103.0, | ||
memory: [20], | ||
}, | ||
"RX 7900 GRE": { | ||
tflops: 91.96, | ||
memory: [16], | ||
}, | ||
"RX 7800 XT": { | ||
tflops: 74.65, | ||
memory: [16], | ||
}, | ||
"RX 7700 XT": { | ||
tflops: 70.34, | ||
memory: [12], | ||
}, | ||
"RX 7600 XT": { | ||
tflops: 45.14, | ||
memory: [16, 8], | ||
}, | ||
}, | ||
}, | ||
CPU: { | ||
Intel: { | ||
"Xeon 4th Generation (Sapphire Rapids)": { | ||
tflops: 1.3, | ||
}, | ||
"Xeon 3th Generation (Ice Lake)": { | ||
tflops: 0.8, | ||
}, | ||
"Xeon 2th Generation (Cascade Lake)": { | ||
tflops: 0.55, | ||
}, | ||
"Intel Core 13th Generation (i9)": { | ||
tflops: 0.85, | ||
}, | ||
"Intel Core 13th Generation (i7)": { | ||
tflops: 0.82, | ||
}, | ||
"Intel Core 13th Generation (i5)": { | ||
tflops: 0.68, | ||
}, | ||
"Intel Core 13th Generation (i3)": { | ||
tflops: 0.57, | ||
}, | ||
"Intel Core 12th Generation (i9)": { | ||
tflops: 0.79, | ||
}, | ||
"Intel Core 12th Generation (i7)": { | ||
tflops: 0.77, | ||
}, | ||
"Intel Core 12th Generation (i5)": { | ||
tflops: 0.65, | ||
}, | ||
"Intel Core 12th Generation (i3)": { | ||
tflops: 0.53, | ||
}, | ||
"Intel Core 11th Generation (i9)": { | ||
tflops: 0.7, | ||
}, | ||
"Intel Core 11th Generation (i7)": { | ||
tflops: 0.6, | ||
}, | ||
"Intel Core 11th Generation (i5)": { | ||
tflops: 0.5, | ||
}, | ||
"Intel Core 11th Generation (i3)": { | ||
tflops: 0.35, | ||
}, | ||
"Intel Core 10th Generation (i9)": { | ||
tflops: 0.46, | ||
}, | ||
"Intel Core 10th Generation (i7)": { | ||
tflops: 0.46, | ||
}, | ||
"Intel Core 10th Generation (i5)": { | ||
tflops: 0.46, | ||
}, | ||
"Intel Core 10th Generation (i3)": { | ||
tflops: 0.44, | ||
}, | ||
}, | ||
AMD: { | ||
"EPYC 4th Generation (Genoa)": { | ||
tflops: 5, | ||
}, | ||
"EPYC 3th Generation (Milan)": { | ||
tflops: 2.4, | ||
}, | ||
"EPYC 2th Generation (Rome)": { | ||
tflops: 0.6, | ||
}, | ||
"EPYC 1st Generation (Naples)": { | ||
tflops: 0.6, | ||
}, | ||
"Ryzen Zen4 7000 (Ryzen 9)": { | ||
tflops: 0.56, | ||
}, | ||
"Ryzen Zen4 7000 (Ryzen 7)": { | ||
tflops: 0.56, | ||
}, | ||
"Ryzen Zen4 7000 (Ryzen 5)": { | ||
tflops: 0.56, | ||
}, | ||
"Ryzen Zen3 5000 (Ryzen 9)": { | ||
tflops: 1.33, | ||
}, | ||
"Ryzen Zen3 5000 (Ryzen 7)": { | ||
tflops: 1.33, | ||
}, | ||
"Ryzen Zen3 5000 (Ryzen 5)": { | ||
tflops: 0.72, | ||
}, | ||
"Ryzen Zen 2 3000 (Threadripper)": { | ||
tflops: 0.72, | ||
}, | ||
"Ryzen Zen 2 3000 (Ryzen 9)": { | ||
tflops: 0.72, | ||
}, | ||
"Ryzen Zen 2 3000 (Ryzen 7)": { | ||
tflops: 0.72, | ||
}, | ||
"Ryzen Zen 2 3000 (Ryzen 5)": { | ||
tflops: 0.72, | ||
}, | ||
"Ryzen Zen 2 3000 (Ryzen 3)": { | ||
tflops: 0.72, | ||
}, | ||
}, | ||
}, | ||
"Apple Silicon": { | ||
"-": { | ||
"Apple M1": { | ||
tflops: 2.6, | ||
memory: [8, 16], | ||
}, | ||
"Apple M1 Pro": { | ||
tflops: 5.2, | ||
memory: [16, 24, 32], | ||
}, | ||
"Apple M1 Max": { | ||
tflops: 10.4, | ||
memory: [16, 24, 32, 64], | ||
}, | ||
"Apple M1 Ultra": { | ||
tflops: 21, | ||
memory: [16, 24, 32, 64, 96, 128], | ||
}, | ||
"Apple M2": { | ||
tflops: 3.6, | ||
memory: [8, 16, 24], | ||
}, | ||
"Apple M2 Pro": { | ||
tflops: 13.6, | ||
memory: [16, 24, 32], | ||
}, | ||
"Apple M2 Max": { | ||
tflops: 13.49, | ||
memory: [32, 64, 96], | ||
}, | ||
"Apple M2 Ultra": { | ||
tflops: 27.2, | ||
memory: [64, 96, 128, 192], | ||
}, | ||
"Apple M3": { | ||
tflops: 2.84, | ||
memory: [8, 16, 24], | ||
}, | ||
"Apple M3 Pro": { | ||
tflops: 14, | ||
memory: [18, 36], | ||
}, | ||
"Apple M3 Max": { | ||
tflops: 14.2, | ||
memory: [36, 48, 64, 96, 128], | ||
}, | ||
}, | ||
}, | ||
} satisfies Record<string, Record<string, Record<string, HardwareSpec>>>; | ||
|
||
export type SkuType = keyof typeof SKUS; |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters