1 | #pragma once |
---|---|
2 | #include <cstdint> |
3 | |
4 | #include <c10/macros/Macros.h> |
5 | |
6 | namespace c10 { |
7 | |
8 | /** |
9 | * quint8 is for unsigned 8 bit quantized Tensors |
10 | */ |
11 | struct alignas(1) quint8 { |
12 | using underlying = uint8_t; |
13 | uint8_t val_; |
14 | quint8() = default; |
15 | C10_HOST_DEVICE explicit quint8(uint8_t val) : val_(val) {} |
16 | }; |
17 | |
18 | } // namespace c10 |
19 |