frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add various uninterpreted bit tensor data types (try 2) (#95860)

Browse Source 
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
This commit is contained in:
vasiliy
2023-03-04 03:35:59 +00:00
committed by PyTorch MergeBot
parent 5e1067bcc2
commit dc70e8175f
6 changed files with 174 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_quantization.py
View File

@ -134,5 +134,8 @@ try:
except ImportError:
pass
# Experimental functionality
from quantization.core.experimental.test_bits import TestBits # noqa: F401
if __name__ == '__main__':
run_tests()