fixed an assert error

6145fd87 · Shucai Xiao · e8705f4d · 6145fd87
Commit 6145fd87 authored Aug 31, 2019 by Shucai Xiao
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

src/quantization.cpp src/quantization.cpp +2 -1

No files found.
--- a/src/quantization.cpp
+++ b/src/quantization.cpp
@@ -42,7 +42,8 @@ instruction_ref insert_quant_ins(program& prog,

    assert(ins->get_shape().type() == shape::float_type or
           ins->get_shape().type() == shape::double_type or
-           ins->get_shape().type() == shape::int32_type);
+           ins->get_shape().type() == shape::int32_type or
+           ins->get_shape().type() == shape::half_type);
    instruction_ref quant_ins{};
    auto insert_loc = std::next(ins);
    if(type == shape::int8_type)