mirror of https://github.com/llvm/torch-mlir
Run format_sources.sh.
parent
b5f010284f
commit
fb895173f2
|
@ -109,7 +109,6 @@ matchAndRewriteUnaryElementwise(Operation *op, PatternRewriter &rewriter) {
|
|||
"unhandled op (see dump above): TCF->TCP unary elementwise");
|
||||
}
|
||||
return success();
|
||||
|
||||
}
|
||||
|
||||
namespace {
|
||||
|
|
|
@ -111,7 +111,6 @@ mlir::NPCOMP::createLowerAllocMemRefOpsPass() {
|
|||
// RestrictedCanonicalizer
|
||||
//===----------------------------------------------------------------------===//
|
||||
|
||||
|
||||
namespace {
|
||||
struct RestrictedCanonicalizer
|
||||
: public RestrictedCanonicalizerBase<RestrictedCanonicalizer> {
|
||||
|
|
|
@ -85,7 +85,7 @@ class LowerConstantTensorsToMemref
|
|||
registry.insert<tcp::TCPDialect>();
|
||||
}
|
||||
|
||||
void runOnOperation () override {
|
||||
void runOnOperation() override {
|
||||
auto module = getOperation();
|
||||
GlobalCreator globals(module);
|
||||
|
||||
|
|
|
@ -140,8 +140,8 @@ static Value createLinalgBodyCalculationForElementwiseOp(Operation *op,
|
|||
return builder.create<AddFOp>(loc, bodyArgs[0], bodyArgs[1]);
|
||||
|
||||
if (isa<tcp::MaxOp>(op)) {
|
||||
auto greater =
|
||||
builder.create<CmpFOp>(loc, CmpFPredicate::OGT, bodyArgs[0], bodyArgs[1]);
|
||||
auto greater = builder.create<CmpFOp>(loc, CmpFPredicate::OGT, bodyArgs[0],
|
||||
bodyArgs[1]);
|
||||
return builder.create<SelectOp>(loc, greater, bodyArgs[0], bodyArgs[1]);
|
||||
}
|
||||
|
||||
|
@ -158,7 +158,7 @@ static Value createLinalgBodyCalculationForElementwiseOp(Operation *op,
|
|||
|
||||
static LogicalResult
|
||||
matchAndRewriteElementwiseOp(Operation *op, ArrayRef<Value> operands,
|
||||
ConversionPatternRewriter &rewriter) {
|
||||
ConversionPatternRewriter &rewriter) {
|
||||
Location loc = op->getLoc();
|
||||
Value result = op->getResult(0);
|
||||
|
||||
|
|
Loading…
Reference in New Issue