Skip to content

Fix for max block dim#2631

Merged
awni merged 1 commit intoml-explore:mainfrom
awni:compiled_max_block_dim
Sep 29, 2025
Merged

Fix for max block dim#2631
awni merged 1 commit intoml-explore:mainfrom
awni:compiled_max_block_dim

Conversation

@awni
Copy link
Copy Markdown
Member

@awni awni commented Sep 28, 2025

Uses the occupancy size as the block dim for compiled kernels to avoid using too large of a block dim.

Closes #2628

Copy link
Copy Markdown
Member

@angeloskath angeloskath left a comment

Choose a reason for hiding this comment

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

🙏

@awni awni merged commit dc371ae into ml-explore:main Sep 29, 2025
7 checks passed
@awni awni deleted the compiled_max_block_dim branch October 13, 2025 14:55
faisalmemon pushed a commit to faisalmemon/mlx that referenced this pull request Oct 30, 2025
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.

[BUG] [CUDA] Compiled kernel fails with resource issue

2 participants