Michael Goin 14f301b541

Update to torch==2.6.0 (#12721) ...

Signed-off-by: mgoin <michael@neuralmagic.com>
Signed-off-by: mgoin <mgoin64@gmail.com>
Signed-off-by: luka <luka@neuralmagic.com>
Signed-off-by: Tyler Michael Smith <tyler@neuralmagic.com>
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
Co-authored-by: Luka Govedič <ProExpertProg@users.noreply.github.com>
Co-authored-by: DarkLight1337 <tlleungac@connect.ust.hk>
Co-authored-by: Tyler Michael Smith <tyler@neuralmagic.com>
Co-authored-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>

2025-03-14 16:58:30 -04:00

..

piecewise

Update deprecated Python 3.8 typing (#13971)

2025-03-02 17:34:51 -08:00

__init__.py

[torch.compile] register allreduce operations as custom ops (#8526)

2024-09-16 22:57:57 -07:00

backend.py

Update to torch==2.6.0 (#12721)

2025-03-14 16:58:30 -04:00

test_basic_correctness.py

Update deprecated Python 3.8 typing (#13971)

2025-03-02 17:34:51 -08:00

test_full_graph.py

[Misc] Add SPDX-License-Identifier headers to python source files (#12628)

2025-02-02 11:58:18 -08:00

test_functionalization.py

[torch.compile] Fix RMSNorm + quant fusion in the non-cutlass-fp8 case, rename RedundantReshapesPass to NoopEliminationPass (#10902)

2025-02-28 16:20:11 -07:00

test_fusion.py

[FP8] Refactor apply_fp8_linear and apply_fp8_linear_generic into an object (#14390)

2025-03-07 05:20:16 +00:00

test_pass_manager.py

[Misc] Add SPDX-License-Identifier headers to python source files (#12628)

2025-02-02 11:58:18 -08:00

test_wrapper.py

[Misc] Add SPDX-License-Identifier headers to python source files (#12628)

2025-02-02 11:58:18 -08:00

utils.py

Consolidate Llama model usage in tests (#13094)

2025-02-13 22:18:03 -08:00