quint2x4.h 366 B

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