pytorch/test/quantization/core/experimental
vasiliy 61fe49b8ed pt2: make aot_eager backend handle basic float8 operations (#107783)
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
2023-08-23 18:10:53 +00:00
..
apot_fx_graph_mode_ptq.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
apot_fx_graph_mode_qat.py [BE] Enable ruff's UP rules and autoformat test/ (#105434) 2023-07-19 20:36:06 +00:00
quantization_util.py
test_bits.py
test_fake_quantize.py
test_float8.py pt2: make aot_eager backend handle basic float8 operations (#107783) 2023-08-23 18:10:53 +00:00
test_linear.py
test_nonuniform_observer.py
test_quantized_tensor.py
test_quantizer.py