quint8.h 320 B

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