33 lines
811 B
C++
33 lines
811 B
C++
#include <cutlass/complex.h>
|
|
|
|
namespace cutlass {
|
|
|
|
/// ENUM class for datatypes
|
|
enum class DataType {
|
|
kB1, kU2, kU4, kU8,
|
|
kU16, kU32, kU64, kS2,
|
|
kS4, kS8, kS16, kS32,
|
|
kS64, kF16, kBF16, kF32,
|
|
kTF32, kF64, kCF16, kCBF16,
|
|
kCF32, kCTF32, kCF64, kCS2,
|
|
kCS4, kCS8, kCS16, kCS32,
|
|
kCS64, kCU2, kCU4, kCU8,
|
|
kCU16, kCU32, kCU64, kInvalid
|
|
};
|
|
|
|
/// ENUM class for LayoutTypes
|
|
enum class LayoutType {
|
|
kColumnMajor, kRowMajor,
|
|
kColumnMajorInterleaved2, kRowMajorInterleaved2,
|
|
kColumnMajorInterleaved32, kRowMajorInterleaved32,
|
|
kColumnMajorInterleaved64, kRowMajorInterleaved64,
|
|
kTensorNHWC, kTensorNDHWC, kTensorNCHW, kTensorNGHWC,
|
|
kTensorNC32HW32, kTensorNC64HW64, kTensorC32RSK32,
|
|
kTensorC64RSK64
|
|
};
|
|
|
|
/// ENUM class for opcode class
|
|
|
|
|
|
} // namespace cutlass
|