Function tryFoldSelectWithSelect

lib/Dialect/CudaTile/IR/CudaTile.cpp:5281–5305

}

static OpFoldResult tryFoldSelectWithSelect(SelectOp op,
                                            SelectOp::FoldAdaptor adaptor) {
  Value t = op.getValIfTrue();
  Value f = op.getValIfFalse();
  Value cond = op.getCond();
  // ---- Rule: select(pred, select(pred, a, b), c) => select(pred, a, c)
  // "RedundantSelectTrue"
  if (auto innerTrueSel = t.getDefiningOp<SelectOp>()) {
    if (innerTrueSel.getCond() == cond) {
      op.getValIfTrueMutable().assign(innerTrueSel.getValIfTrue());
      return op.getResult(); // in-place
    }
  }

  // ---- Rule: select(pred, a, select(pred, b, c)) => select(pred, a, c)
  // "RedundantSelectFalse"
  if (auto innerFalseSel = f.getDefiningOp<SelectOp>()) {
    if (innerFalseSel.getCond() == cond) {
      op.getValIfFalseMutable().assign(innerFalseSel.getValIfFalse());
      return op.getResult();
    }
  }

  return {};
}

namespace {
using SelectFoldRuleFn = OpFoldResult (*)(SelectOp, SelectOp::FoldAdaptor);