|
1 | 1 | """MODULE.bazel extensions for Mojo toolchains."""
|
2 | 2 |
|
3 |
| -_PLATFORMS = ["linux_aarch64", "linux_x86_64", "macosx_13_0_arm64"] |
4 |
| -_DEFAULT_VERSION = "25.4.0.dev2025050605" |
| 3 | +_PLATFORMS = ["linux_aarch64", "linux_x86_64", "macos_arm64"] |
| 4 | +_DEFAULT_VERSION = "25.4.0.dev2025050902" |
5 | 5 | _KNOWN_SHAS = {
|
6 |
| - "25.4.0.dev2025050605": { |
7 |
| - "linux_aarch64": "77acfb83a6d9286c79791e6a443f5160b92dbd5aa69fc370b20f8ca76216100a", |
8 |
| - "linux_x86_64": "35bb85e0101ce0d06eb6731fc18b29d74e15908fb3435465202eb923b80f5f6d", |
9 |
| - "macosx_13_0_arm64": "98f32b413fa7755bbd612c394647c09dea92810856b3f823ab362f8e42ece110", |
| 6 | + "25.4.0.dev2025050902": { |
| 7 | + "linux_aarch64": "d52c67f245575397d8176010d27bd12e76cde297ed8ee7f07dcc73fe48955508", |
| 8 | + "linux_x86_64": "69898a4ffb328489e5c7c1c7e0cba37cd64dd0fa87b4a98501b3562dc89f2695", |
| 9 | + "macos_arm64": "8856745cab1cb88fbba174afb9784cbdda865c8a4e4db5693750efefe7505160", |
10 | 10 | },
|
11 | 11 | }
|
| 12 | +_PLATFORM_MAPPINGS = { |
| 13 | + "linux_aarch64": "manylinux_2_34_aarch64", |
| 14 | + "linux_x86_64": "manylinux_2_34_x86_64", |
| 15 | + "macos_arm64": "macosx_13_0_arm64", |
| 16 | +} |
12 | 17 |
|
13 | 18 | def _mojo_toolchain_impl(rctx):
|
14 | 19 | rctx.download_and_extract(
|
15 |
| - url = "https://dl.modular.com/public/max-nightly/python/max-{}-py3-none-{}.whl".format( |
| 20 | + url = "https://dl.modular.com/public/nightly/python/max-{}-py3-none-{}.whl".format( |
16 | 21 | rctx.attr.version,
|
17 |
| - rctx.attr.platform, |
| 22 | + _PLATFORM_MAPPINGS[rctx.attr.platform], |
18 | 23 | ),
|
19 | 24 | sha256 = _KNOWN_SHAS[rctx.attr.version][rctx.attr.platform],
|
20 | 25 | type = "zip",
|
@@ -68,7 +73,7 @@ toolchain(
|
68 | 73 | """.format(
|
69 | 74 | platform = platform,
|
70 | 75 | cpu = "x86_64" if "x86_64" in platform else "aarch64",
|
71 |
| - os = "macos" if "macosx" in platform else "linux", |
| 76 | + os = "macos" if "macos" in platform else "linux", |
72 | 77 | ))
|
73 | 78 |
|
74 | 79 | rctx.file("BUILD.bazel", content = "\n".join(lines))
|
|
0 commit comments