Skip to content

Update pixi.toml to latest MAX release, add support for aarch64 Linux#227

Merged
dunnoconnor merged 5 commits intomodular:mainfrom
BradLarson:aarch64
Mar 5, 2026
Merged

Update pixi.toml to latest MAX release, add support for aarch64 Linux#227
dunnoconnor merged 5 commits intomodular:mainfrom
BradLarson:aarch64

Conversation

@BradLarson
Copy link
Collaborator

The latest nightly version of MAX adds full support for running the GPU puzzles on DGX Spark and Jetson devices. These systems are aarch64 Linux hosts, so we also need to add linux-aarch64 entries. I believe PyTorch only started supporting aarch64 for CUDA in 2.9.0, so that version is used for the imports on this architecture. PyTorch properly installs and tests pass on DGX Spark at that version, and not for the older import used in linux-64.

All tests pass on DGX Spark except for the last three, which will need upstream kernel enhancements to reflect this new architecture. They should be supported on this device, but we haven't yet added the appropriate checks yet.

Additionally, there are some significant Mojo API changes that this nightly starts to support. The migration to those will come in a follow-on PR to keep this one focused.

@BradLarson
Copy link
Collaborator Author

Hmm, we need to set MODULAR_NVPTX_COMPILER_PATH for the CI system to work correctly our current nightlies, but the CI systems don't appear to have ptxas installed or at least not installed in the normal location. More investigation may be needed.

@dunnoconnor dunnoconnor merged commit 6ce4d81 into modular:main Mar 5, 2026
2 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants