Skip to content

[metal] Wire @metal_jit into MetalBackend and simplify launcher#1992

Open
aditvenk wants to merge 1 commit intomainfrom
aditvenk/stack/26
Open

[metal] Wire @metal_jit into MetalBackend and simplify launcher#1992
aditvenk wants to merge 1 commit intomainfrom
aditvenk/stack/26

Conversation

@aditvenk
Copy link
Copy Markdown
Contributor

@aditvenk aditvenk commented Apr 9, 2026

Stacked PRs:


[metal] Wire @metal_jit into MetalBackend and simplify launcher

  • MetalBackend.function_decorator returns "metal_jit"
  • Add metal_jit to library_imports
  • Launcher passes *args through to metal_kernel for dtype inference
  • Add copy kernel, elementwise, and codegen inspection tests

@aditvenk aditvenk force-pushed the aditvenk/stack/26 branch from d54f8aa to df56ec5 Compare April 9, 2026 04:50
@aditvenk aditvenk force-pushed the aditvenk/stack/25 branch from 276e63c to 951fddc Compare April 9, 2026 04:50
@meta-cla meta-cla bot added the CLA Signed This label is managed by the Meta Open Source bot. label Apr 9, 2026
@aditvenk aditvenk marked this pull request as draft April 9, 2026 20:43
@aditvenk aditvenk changed the base branch from aditvenk/stack/25 to main April 9, 2026 20:43
@aditvenk aditvenk force-pushed the aditvenk/stack/26 branch from df56ec5 to 4772e33 Compare April 9, 2026 20:43
@aditvenk aditvenk changed the base branch from main to aditvenk/stack/25 April 9, 2026 20:43
@aditvenk aditvenk marked this pull request as ready for review April 9, 2026 20:43
@aditvenk aditvenk marked this pull request as draft April 9, 2026 20:57
@aditvenk aditvenk changed the base branch from aditvenk/stack/25 to main April 9, 2026 20:57
@aditvenk aditvenk force-pushed the aditvenk/stack/26 branch from 4772e33 to 13025df Compare April 9, 2026 20:57
@aditvenk aditvenk changed the base branch from main to aditvenk/stack/25 April 9, 2026 20:58
@aditvenk aditvenk marked this pull request as ready for review April 9, 2026 20:58
@aditvenk aditvenk marked this pull request as draft April 9, 2026 21:50
@aditvenk aditvenk changed the base branch from aditvenk/stack/25 to main April 9, 2026 21:50
@aditvenk aditvenk force-pushed the aditvenk/stack/26 branch from 13025df to 103e58e Compare April 9, 2026 21:50
@aditvenk aditvenk changed the base branch from main to aditvenk/stack/25 April 9, 2026 21:50
@aditvenk aditvenk marked this pull request as ready for review April 9, 2026 21:50
@aditvenk aditvenk marked this pull request as draft April 13, 2026 02:14
@aditvenk aditvenk changed the base branch from aditvenk/stack/25 to main April 13, 2026 02:14
@aditvenk aditvenk force-pushed the aditvenk/stack/26 branch from 103e58e to 2dd7d1d Compare April 13, 2026 02:14
@aditvenk aditvenk changed the base branch from main to aditvenk/stack/25 April 13, 2026 02:14
@aditvenk aditvenk marked this pull request as ready for review April 13, 2026 02:15
- MetalBackend.function_decorator returns "metal_jit"
- Add metal_jit to library_imports
- Launcher passes *args through to metal_kernel for dtype inference
- Add copy kernel, elementwise, and codegen inspection tests

stack-info: PR: #1992, branch: aditvenk/stack/26
@aditvenk aditvenk marked this pull request as draft April 13, 2026 02:54
@aditvenk aditvenk changed the base branch from aditvenk/stack/25 to main April 13, 2026 02:54
@aditvenk aditvenk force-pushed the aditvenk/stack/26 branch from 2dd7d1d to db1db7e Compare April 13, 2026 02:54
@aditvenk aditvenk marked this pull request as ready for review April 13, 2026 02:54
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Meta Open Source bot.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants