Skip to content

Commit

Permalink
Add missing overload
Browse files Browse the repository at this point in the history
  • Loading branch information
abadams committed Jul 10, 2023
1 parent 81efe39 commit 3ac8bc1
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions src/IROperator.h
Original file line number Diff line number Diff line change
Expand Up @@ -826,7 +826,13 @@ inline Tuple select(const Expr &c0, const Tuple &v0, const Expr &c1, const Tuple
* return an Expr. A runtime error is produced if this is applied to
* tuple-valued Funcs. In that case you should explicitly cast the second and
* third args to Tuple to remove the ambiguity. */
// @{
Expr select(const Expr &condition, const FuncRef &true_value, const FuncRef &false_value);
template<typename... Args>
inline Expr select(const Expr &c0, const FuncRef &v0, const Expr &c1, const FuncRef &v1, Args &&...args) {
return select(c0, v0, select(c1, v1, std::forward<Args>(args)...));
}
// @}

/** Oftentimes we want to pack a list of expressions with the same type
* into a channel dimension, e.g.,
Expand Down

0 comments on commit 3ac8bc1

Please sign in to comment.