Commit 08e4d085 by eqy Committed by Tianqi Chen

remove minimum 32-bit restriction (#621)

Change minimum 32-bit restriction for floating point types to 8-bit.
This change is to enable reduced precision types that may use vector operations underneath the hood (cases #lanes > 1 such as half4).
parent c7101537
......@@ -127,7 +127,7 @@ inline void TVMArrayFree_(TVMArray* arr) {
inline void VerifyType(int dtype_code, int dtype_bits, int dtype_lanes) {
CHECK_GE(dtype_lanes, 1);
if (dtype_code == kDLFloat) {
CHECK_EQ(dtype_bits % 32, 0);
CHECK_EQ(dtype_bits % 8, 0);
} else {
CHECK_EQ(dtype_bits % 8, 0);
}
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment