alloc 0.0.1__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- alloc/__init__.py +11 -0
- alloc/artifact_writer.py +67 -0
- alloc/callbacks.py +342 -0
- alloc/catalog/__init__.py +138 -0
- alloc/catalog/default_rate_card.json +18 -0
- alloc/catalog/gpus.v1.json +174 -0
- alloc/cli.py +1341 -0
- alloc/config.py +124 -0
- alloc/context.py +191 -0
- alloc/display.py +580 -0
- alloc/extractor_runner.py +141 -0
- alloc/ghost.py +167 -0
- alloc/model_extractor.py +170 -0
- alloc/model_registry.py +138 -0
- alloc/probe.py +461 -0
- alloc/stability.py +144 -0
- alloc/upload.py +138 -0
- alloc/yaml_config.py +287 -0
- alloc-0.0.1.dist-info/METADATA +256 -0
- alloc-0.0.1.dist-info/RECORD +23 -0
- alloc-0.0.1.dist-info/WHEEL +5 -0
- alloc-0.0.1.dist-info/entry_points.txt +2 -0
- alloc-0.0.1.dist-info/top_level.txt +1 -0
|
@@ -0,0 +1,174 @@
|
|
|
1
|
+
{
|
|
2
|
+
"version": "1",
|
|
3
|
+
"gpus": {
|
|
4
|
+
"nvidia-h200-141gb": {
|
|
5
|
+
"display_name": "H200",
|
|
6
|
+
"vendor": "nvidia",
|
|
7
|
+
"vram_gb": 141,
|
|
8
|
+
"bandwidth_gbps": 4800,
|
|
9
|
+
"fp16_tflops": 989,
|
|
10
|
+
"bf16_tflops": 989,
|
|
11
|
+
"fp32_tflops": 67,
|
|
12
|
+
"tf32_tflops": 495,
|
|
13
|
+
"architecture": "Hopper",
|
|
14
|
+
"tdp_watts": 700,
|
|
15
|
+
"interconnect": { "nvlink_gen": 4, "nvlink_bw_gbps": 900, "pcie_gen": 5 }
|
|
16
|
+
},
|
|
17
|
+
"nvidia-h100-sxm-80gb": {
|
|
18
|
+
"display_name": "H100-80GB",
|
|
19
|
+
"vendor": "nvidia",
|
|
20
|
+
"vram_gb": 80,
|
|
21
|
+
"bandwidth_gbps": 3350,
|
|
22
|
+
"fp16_tflops": 989,
|
|
23
|
+
"bf16_tflops": 989,
|
|
24
|
+
"fp32_tflops": 67,
|
|
25
|
+
"tf32_tflops": 495,
|
|
26
|
+
"architecture": "Hopper",
|
|
27
|
+
"tdp_watts": 700,
|
|
28
|
+
"interconnect": { "nvlink_gen": 4, "nvlink_bw_gbps": 900, "pcie_gen": 5 }
|
|
29
|
+
},
|
|
30
|
+
"nvidia-h100-nvl-94gb": {
|
|
31
|
+
"display_name": "H100-NVL",
|
|
32
|
+
"vendor": "nvidia",
|
|
33
|
+
"vram_gb": 94,
|
|
34
|
+
"bandwidth_gbps": 3350,
|
|
35
|
+
"fp16_tflops": 989,
|
|
36
|
+
"bf16_tflops": 989,
|
|
37
|
+
"fp32_tflops": 67,
|
|
38
|
+
"tf32_tflops": 495,
|
|
39
|
+
"architecture": "Hopper",
|
|
40
|
+
"tdp_watts": 400,
|
|
41
|
+
"interconnect": { "nvlink_gen": 4, "nvlink_bw_gbps": 900, "pcie_gen": 5 }
|
|
42
|
+
},
|
|
43
|
+
"nvidia-a100-sxm-80gb": {
|
|
44
|
+
"display_name": "A100-80GB",
|
|
45
|
+
"vendor": "nvidia",
|
|
46
|
+
"vram_gb": 80,
|
|
47
|
+
"bandwidth_gbps": 2039,
|
|
48
|
+
"fp16_tflops": 312,
|
|
49
|
+
"bf16_tflops": 312,
|
|
50
|
+
"fp32_tflops": 19.5,
|
|
51
|
+
"tf32_tflops": 156,
|
|
52
|
+
"architecture": "Ampere",
|
|
53
|
+
"tdp_watts": 400,
|
|
54
|
+
"interconnect": { "nvlink_gen": 3, "nvlink_bw_gbps": 600, "pcie_gen": 4 }
|
|
55
|
+
},
|
|
56
|
+
"nvidia-a100-40gb": {
|
|
57
|
+
"display_name": "A100-40GB",
|
|
58
|
+
"vendor": "nvidia",
|
|
59
|
+
"vram_gb": 40,
|
|
60
|
+
"bandwidth_gbps": 1555,
|
|
61
|
+
"fp16_tflops": 312,
|
|
62
|
+
"bf16_tflops": 312,
|
|
63
|
+
"fp32_tflops": 19.5,
|
|
64
|
+
"tf32_tflops": 156,
|
|
65
|
+
"architecture": "Ampere",
|
|
66
|
+
"tdp_watts": 400,
|
|
67
|
+
"interconnect": { "nvlink_gen": 3, "nvlink_bw_gbps": 600, "pcie_gen": 4 }
|
|
68
|
+
},
|
|
69
|
+
"nvidia-a10g-24gb": {
|
|
70
|
+
"display_name": "A10G",
|
|
71
|
+
"vendor": "nvidia",
|
|
72
|
+
"vram_gb": 24,
|
|
73
|
+
"bandwidth_gbps": 600,
|
|
74
|
+
"fp16_tflops": 125,
|
|
75
|
+
"bf16_tflops": 125,
|
|
76
|
+
"fp32_tflops": 31.2,
|
|
77
|
+
"tf32_tflops": 62.5,
|
|
78
|
+
"architecture": "Ampere",
|
|
79
|
+
"tdp_watts": 300,
|
|
80
|
+
"interconnect": { "pcie_gen": 4 }
|
|
81
|
+
},
|
|
82
|
+
"nvidia-l40s-48gb": {
|
|
83
|
+
"display_name": "L40S",
|
|
84
|
+
"vendor": "nvidia",
|
|
85
|
+
"vram_gb": 48,
|
|
86
|
+
"bandwidth_gbps": 864,
|
|
87
|
+
"fp16_tflops": 362,
|
|
88
|
+
"bf16_tflops": 362,
|
|
89
|
+
"fp32_tflops": 91.6,
|
|
90
|
+
"tf32_tflops": 183,
|
|
91
|
+
"architecture": "Ada Lovelace",
|
|
92
|
+
"tdp_watts": 350,
|
|
93
|
+
"interconnect": { "pcie_gen": 4 }
|
|
94
|
+
},
|
|
95
|
+
"nvidia-l4-24gb": {
|
|
96
|
+
"display_name": "L4",
|
|
97
|
+
"vendor": "nvidia",
|
|
98
|
+
"vram_gb": 24,
|
|
99
|
+
"bandwidth_gbps": 300,
|
|
100
|
+
"fp16_tflops": 121,
|
|
101
|
+
"bf16_tflops": 121,
|
|
102
|
+
"fp32_tflops": 30.3,
|
|
103
|
+
"tf32_tflops": 60,
|
|
104
|
+
"architecture": "Ada Lovelace",
|
|
105
|
+
"tdp_watts": 72,
|
|
106
|
+
"interconnect": { "pcie_gen": 4 }
|
|
107
|
+
},
|
|
108
|
+
"nvidia-t4-16gb": {
|
|
109
|
+
"display_name": "T4",
|
|
110
|
+
"vendor": "nvidia",
|
|
111
|
+
"vram_gb": 16,
|
|
112
|
+
"bandwidth_gbps": 320,
|
|
113
|
+
"fp16_tflops": 65,
|
|
114
|
+
"bf16_tflops": 0,
|
|
115
|
+
"fp32_tflops": 8.1,
|
|
116
|
+
"tf32_tflops": 0,
|
|
117
|
+
"architecture": "Turing",
|
|
118
|
+
"tdp_watts": 70,
|
|
119
|
+
"interconnect": { "pcie_gen": 3 }
|
|
120
|
+
},
|
|
121
|
+
"nvidia-v100-32gb": {
|
|
122
|
+
"display_name": "V100-32GB",
|
|
123
|
+
"vendor": "nvidia",
|
|
124
|
+
"vram_gb": 32,
|
|
125
|
+
"bandwidth_gbps": 900,
|
|
126
|
+
"fp16_tflops": 125,
|
|
127
|
+
"bf16_tflops": 0,
|
|
128
|
+
"fp32_tflops": 15.7,
|
|
129
|
+
"tf32_tflops": 0,
|
|
130
|
+
"architecture": "Volta",
|
|
131
|
+
"tdp_watts": 300,
|
|
132
|
+
"interconnect": { "nvlink_gen": 2, "nvlink_bw_gbps": 300, "pcie_gen": 3 }
|
|
133
|
+
},
|
|
134
|
+
"nvidia-v100-16gb": {
|
|
135
|
+
"display_name": "V100-16GB",
|
|
136
|
+
"vendor": "nvidia",
|
|
137
|
+
"vram_gb": 16,
|
|
138
|
+
"bandwidth_gbps": 900,
|
|
139
|
+
"fp16_tflops": 125,
|
|
140
|
+
"bf16_tflops": 0,
|
|
141
|
+
"fp32_tflops": 15.7,
|
|
142
|
+
"tf32_tflops": 0,
|
|
143
|
+
"architecture": "Volta",
|
|
144
|
+
"tdp_watts": 300,
|
|
145
|
+
"interconnect": { "nvlink_gen": 2, "nvlink_bw_gbps": 300, "pcie_gen": 3 }
|
|
146
|
+
},
|
|
147
|
+
"nvidia-rtx4090-24gb": {
|
|
148
|
+
"display_name": "RTX-4090",
|
|
149
|
+
"vendor": "nvidia",
|
|
150
|
+
"vram_gb": 24,
|
|
151
|
+
"bandwidth_gbps": 1008,
|
|
152
|
+
"fp16_tflops": 330,
|
|
153
|
+
"bf16_tflops": 330,
|
|
154
|
+
"fp32_tflops": 82.6,
|
|
155
|
+
"tf32_tflops": 165,
|
|
156
|
+
"architecture": "Ada Lovelace",
|
|
157
|
+
"tdp_watts": 450,
|
|
158
|
+
"interconnect": { "pcie_gen": 4 }
|
|
159
|
+
},
|
|
160
|
+
"nvidia-rtx3090-24gb": {
|
|
161
|
+
"display_name": "RTX-3090",
|
|
162
|
+
"vendor": "nvidia",
|
|
163
|
+
"vram_gb": 24,
|
|
164
|
+
"bandwidth_gbps": 936,
|
|
165
|
+
"fp16_tflops": 142,
|
|
166
|
+
"bf16_tflops": 142,
|
|
167
|
+
"fp32_tflops": 35.6,
|
|
168
|
+
"tf32_tflops": 71,
|
|
169
|
+
"architecture": "Ampere",
|
|
170
|
+
"tdp_watts": 350,
|
|
171
|
+
"interconnect": { "pcie_gen": 4 }
|
|
172
|
+
}
|
|
173
|
+
}
|
|
174
|
+
}
|