[OpenCL] Avoid SelectNode ambiguous overloading

apache · May 27, 2022 · 60f68d2 · 60f68d2
1 parent 4a769c1
commit 60f68d2
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 13 deletions.
diff --git a/src/target/source/codegen_opencl.cc b/src/target/source/codegen_opencl.cc
@@ -540,16 +540,6 @@ void CodeGenOpenCL::VisitExpr_(const OrNode* op, std::ostream& os) {
   os << ")";
 }
 
-void CodeGenOpenCL::VisitExpr_(const SelectNode* op, std::ostream& os) {
-  os << "select(";
-  PrintExpr(op->false_value, os);
-  os << ", ";
-  PrintExpr(op->true_value, os);
-  os << ", ";
-  PrintExpr(op->condition, os);
-  os << ")";
-}
-
 void CodeGenOpenCL::SetTextureScope(
     const std::unordered_map<const VarNode*, std::string>& scope) {  // NOLINT(*)
   for (auto& texture : scope) {

diff --git a/src/target/source/codegen_opencl.h b/src/target/source/codegen_opencl.h
@@ -72,7 +72,6 @@ class CodeGenOpenCL final : public CodeGenC {
   void VisitExpr_(const MaxNode* op, std::ostream& os) final;
   void VisitExpr_(const AndNode* op, std::ostream& os) final;
   void VisitExpr_(const OrNode* op, std::ostream& os) final;
-  void VisitExpr_(const SelectNode* op, std::ostream& os) final;
 
  private:
   // whether enable fp16 and fp64 extension

diff --git a/tests/python/unittest/test_target_codegen_opencl.py b/tests/python/unittest/test_target_codegen_opencl.py
@@ -173,8 +173,8 @@ def check_type_casting(ctx, n, dtype):
         lcond = "(convert_uint4(((uint4)((((int)get_local_id(0)) == 3), (((int)get_local_id(0)) == 3), (((int)get_local_id(0)) == 3), (((int)get_local_id(0)) == 3)))))"
         rcond = "(convert_uint4((((int4)((0)+(1*0), (0)+(1*1), (0)+(1*2), (0)+(1*3))) == ((int4)(3, 3, 3, 3)))))"
         cond = "({} && {})".format(lcond, rcond)
-        select = "select({}, {}, {})".format(false_branch, true_branch, cond)
-        count = assembly.count(select)
+        ternary = "{} ? {} : {}".format(cond, true_branch, false_branch)
+        count = assembly.count(ternary)
         assert count == 1
 
         fun(c)