Skip to content

Integrate test-backed RMSNorm dispatch frontier - #1896

Merged
NripeshN merged 3 commits into
mainfrom
mlx-rms-norm-test-dispatch
Aug 20, 2026
Merged

Integrate test-backed RMSNorm dispatch frontier#1896
NripeshN merged 3 commits into
mainfrom
mlx-rms-norm-test-dispatch

Conversation

@NripeshN

Copy link
Copy Markdown
Member

Summary

Integrates the pinned MLX RMSNorm forward and gradient test dispatches into the DirectX project-porting frontier. The project harness now emits and validates 12 entry-scoped HLSL artifacts derived from the checked-in host dispatch contract.

Related Issue

Refs #1793

Support issue traceability: no issue closed

Changes

  • Add a deterministic RMSNorm host dispatch contract for the pinned MLX revision.
  • Cover the forward float32, float16, and bfloat16 paths plus weighted and unweighted VJP paths at the single-row and looped workgroup sizes exercised by the pinned tests.
  • Compile all 12 generated artifacts with DXC cs_6_6, native 16-bit types, and warnings as errors on Windows CI.
  • Update frontier accounting to 19 translated DirectX artifacts and 482 compiler runs, leaving four sources and 82 entries fail-closed pending bounded dispatch contracts.
  • Record source, workload, dispatch, subgroup, specialization, and artifact provenance without claiming runtime execution or numerical parity.

Testing

  • Full suite: 17306 passed, 225 skipped
  • Focused MLX and workflow suite: 229 passed
  • Pinned reduced-frontier replay completed successfully.
  • All 12 RMSNorm artifacts compiled with DXC v1.9.2602.24.
  • Support matrix check passed.
  • Support issue sync dry run completed.
  • pre-commit run --all-files passed.

Checklist

  • Tests cover new or changed code
  • Added Support issue traceability: no issue closed
  • Only touched files required for this integration
  • Everything builds and runs locally

Add a pinned host-dispatch contract for the RMSNorm forward and gradient
workloads exercised by the MLX fast-operation tests. Translate the resulting
12 DirectX artifacts independently, retain their dispatch provenance, and
compile every artifact in the Windows project-porting gate.

Update the repository frontier accounting, evidence, documentation, and
focused tests while preserving the explicit boundary between compiler
validation and runtime or numerical parity.
@NripeshN
NripeshN requested a review from vaatsalya123 as a code owner August 13, 2026 11:34
@NripeshN
NripeshN merged commit bafb994 into main Aug 20, 2026
435 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.

1 participant