From f6aede87d3af25d5a2363db329762747073c0510 Mon Sep 17 00:00:00 2001 From: xndcn Date: Wed, 31 Mar 2021 23:12:20 +0800 Subject: [PATCH] [CodeGen][OpenCL] Limit OpenCL built-in vector lanes to 2, 3, 4, 8, 16. --- src/target/source/codegen_opencl.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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; }