pytorch/test/quantization/core
vasiliy dc70e8175f Add various uninterpreted bit tensor data types (try 2) (#95860)
Summary:

This is a retry of https://github.com/pytorch/pytorch/pull/94992 which was reverted due to CI issues.

This PR adds a set of unintrepreted data types on PyTorch which can be used to implement experimental functionality out of core (think fp8, int4, int16 quant, etc).

@bypass-github-export-checks

Test Plan:

```
python test/test_quantization.py -k TestBits
```

Reviewers:

Subscribers:

Tasks:

Tags:

Fixes #ISSUE_NUMBER

Pull Request resolved: https://github.com/pytorch/pytorch/pull/95860
Approved by: https://github.com/atalman
2023-03-04 03:35:59 +00:00
..
experimental Add various uninterpreted bit tensor data types (try 2) (#95860) 2023-03-04 03:35:59 +00:00
__init__.py
test_backend_config.py AO migration: replace torch internal callsites (#94170) 2023-02-07 02:32:23 +00:00
test_docs.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_quantized_functional.py
test_quantized_module.py [quant] add serialization method for quantized hardswish (#94486) 2023-02-24 04:43:27 +00:00
test_quantized_op.py equal_quantized_cpu requires both inputs are quantized tensor (#95875) 2023-03-03 05:33:23 +00:00
test_quantized_tensor.py [quant][pt2e] Add support for dynamic quantization with symmetric quant for input (#94854) 2023-02-28 19:39:31 +00:00
test_top_level_apis.py
test_utils.py AO migration: replace torch internal callsites (#94170) 2023-02-07 02:32:23 +00:00
test_workflow_module.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00
test_workflow_ops.py [BE] [3/3] Rewrite super() calls in test (#94592) 2023-02-12 22:20:53 +00:00