mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
Summary: Reland of https://github.com/pytorch/pytorch/pull/107642 with a fix for tests on Windows. Makes aot_eager backend of torch.compile handle basic float8 operations. This is useful for float8 training UX. Test Plan: ``` python test/test_quantization.py -k test_pt2_traceable_aot_eager ``` Reviewers: Subscribers: Tasks: Tags: Fixes #ISSUE_NUMBER Pull Request resolved: https://github.com/pytorch/pytorch/pull/107783 Approved by: https://github.com/albanD |
||
|---|---|---|
| .. | ||
| apot_fx_graph_mode_ptq.py | ||
| apot_fx_graph_mode_qat.py | ||
| quantization_util.py | ||
| test_bits.py | ||
| test_fake_quantize.py | ||
| test_float8.py | ||
| test_linear.py | ||
| test_nonuniform_observer.py | ||
| test_quantized_tensor.py | ||
| test_quantizer.py | ||