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