21 lines
472 B
C++
21 lines
472 B
C++
#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
|