Skip to content

[CUDA] Disable managed memory on Tegra#3701

Merged
zcbenz merged 2 commits into
ml-explore:mainfrom
wendylabsinc:gab/pr/mem-model
Jun 20, 2026
Merged

[CUDA] Disable managed memory on Tegra#3701
zcbenz merged 2 commits into
ml-explore:mainfrom
wendylabsinc:gab/pr/mem-model

Conversation

@gmondada

Copy link
Copy Markdown
Contributor

Proposed changes

Currently, on Windows and WSL, managed memory support is disabled when concurrentManagedAccess = 0.
This PR wants to disable it on all platforms where concurrentManagedAccess = 0, not just on Windows.
This addresses segmentation fault problems seen on NVIDIA Tegra. See #3645.
In fact, Windows, WSL, and Tegra share the same unified memory paradigm called "Limited unified memory support", which is identified by concurrentManagedAccess = 0.
See

Checklist

Put an x in the boxes that apply.

  • I have read the CONTRIBUTING document
  • I have run pre-commit run --all-files to format my code / installed pre-commit prior to committing changes
  • I have added tests that prove my fix is effective or that my feature works
  • I have updated the necessary documentation (if needed)

@gmondada gmondada marked this pull request as ready for review June 16, 2026 08:39
@gmondada

Copy link
Copy Markdown
Contributor Author

@Joannis 👆

Comment thread mlx/backend/cuda/allocator.cpp Outdated

@zcbenz zcbenz left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks!

@zcbenz zcbenz merged commit 5f5ce2a into ml-explore:main Jun 20, 2026
16 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