| #pragma once | |
| #include <cstdint> | |
| #include <c10/macros/Macros.h> | |
| namespace c10 { | |
| /** | |
| * This is the data type for quantized Tensors. Right now we only have | |
| * qint8 which is for 8 bit Tensors, and qint32 for 32 bit int Tensors, | |
| * we might have 4 bit, 2 bit or 1 bit data types in the future. | |
| */ | |
| struct alignas(1) qint8 { | |
| using underlying = int8_t; | |
| int8_t val_; | |
| qint8() = default; | |
| C10_HOST_DEVICE explicit qint8(int8_t val) : val_(val) {} | |
| }; | |
| } // namespace c10 |