| 1234567891011121314151617181920 |
- #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
|
