PyTorch reference script had wrong concat order (forward) vs the original nikopueringer/CorridorKey which uses torch.cat([c4,c3,c2,c1]). Golden fixtures regenerated. Relax 3 tight e2e tolerances for Metal vs CPU float32 drift. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com> |
||
|---|---|---|
| .. | ||
| bench_mlx.py | ||
| compare_reference.py | ||
| convert_weights.py | ||
| dump_pytorch_reference.py | ||
| infer_pytorch.py | ||
| infer.py | ||
| smoke_2048.py | ||
| smoke_engine.py | ||