fix(metal): gate NAX to gen-18+ (M5/gen-17 miscompute) + regression test - #6

Merged

bri-prism merged 2 commits into

v0.31.6_prismfrom

fix/nax-gate-gen18

Jul 6, 2026

Commits on Jul 6, 2026

fix(metal): gate NAX to gen-18+ in flattened device header (M5/gen-17 miscompute)
bri-prism
committed
test: use class setUp() for default-device config (match MLXTests suite convention)
bri-prism
committed