We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 2b7cb0d commit c98800aCopy full SHA for c98800a
mojo/extensions.bzl
@@ -204,6 +204,10 @@ _gpu_toolchains_tag = tag_class(
204
"mi300x": "amdgpu:gfx942",
205
"mi325": "amdgpu:gfx942",
206
"rtx5090": "nvidia:120a",
207
+ "m1": "metal:10",
208
+ "m2": "metal:20",
209
+ "m3": "metal:30",
210
+ "m4": "metal:40",
211
},
212
doc = "The GPUs supported by this toolchain, mapping to Mojo's target accelerators.",
213
),
@@ -227,6 +231,10 @@ _gpu_toolchains_tag = tag_class(
227
231
"MI325": "mi325",
228
232
"Navi": "radeon",
229
233
"AMD Radeon Graphics": "radeon",
234
+ "Apple M1": "m1",
235
+ "Apple M2": "m2",
236
+ "Apple M3": "m3",
237
+ "Apple M4": "m4",
230
238
239
doc = "The output from nvidia-smi or rocm-smi to the corresponding GPU name in SUPPORTED_GPUS.",
240
0 commit comments