diff --git a/src/target/source/codegen_opencl.cc b/src/target/source/codegen_opencl.cc index 0f79df37701c..f72f3f265511 100644 --- a/src/target/source/codegen_opencl.cc +++ b/src/target/source/codegen_opencl.cc @@ -120,7 +120,7 @@ void CodeGenOpenCL::PrintType(DataType t, std::ostream& os) { // NOLINT(*) break; } if (!fail && lanes == 1) return; - if (!fail && (lanes >= 2 && lanes <= 16)) { + if (!fail && ((lanes >= 2 && lanes <= 4) || lanes == 8 || lanes == 16)) { os << lanes; return; } @@ -154,7 +154,7 @@ void CodeGenOpenCL::PrintType(DataType t, std::ostream& os) { // NOLINT(*) break; } if (!fail && lanes == 1) return; - if (!fail && (lanes >= 2 && lanes <= 16)) { + if (!fail && ((lanes >= 2 && lanes <= 4) || lanes == 8 || lanes == 16)) { os << lanes; return; }