5dce74817b
Return poison from foldBinary/unary if argument(s) is poison. Add ub dialect as dependency to affected dialects (arith, math, spirv, shape). Add poison materialization to dialects. Add tests for some ops from each dialect. Not all affected ops are covered as it will involve a huge copypaste. Differential Revision: https://reviews.llvm.org/D159013
58 lines
2.0 KiB
C++
58 lines
2.0 KiB
C++
//===- ArithDialect.cpp - MLIR Arith dialect implementation -----===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/Dialect/Arith/IR/Arith.h"
|
|
#include "mlir/Dialect/UB/IR/UBOps.h"
|
|
#include "mlir/IR/Builders.h"
|
|
#include "mlir/IR/DialectImplementation.h"
|
|
#include "mlir/Transforms/InliningUtils.h"
|
|
#include "llvm/ADT/TypeSwitch.h"
|
|
|
|
using namespace mlir;
|
|
using namespace mlir::arith;
|
|
|
|
#include "mlir/Dialect/Arith/IR/ArithOpsDialect.cpp.inc"
|
|
#include "mlir/Dialect/Arith/IR/ArithOpsInterfaces.cpp.inc"
|
|
#define GET_ATTRDEF_CLASSES
|
|
#include "mlir/Dialect/Arith/IR/ArithOpsAttributes.cpp.inc"
|
|
|
|
namespace {
|
|
/// This class defines the interface for handling inlining for arithmetic
|
|
/// dialect operations.
|
|
struct ArithInlinerInterface : public DialectInlinerInterface {
|
|
using DialectInlinerInterface::DialectInlinerInterface;
|
|
|
|
/// All arithmetic dialect ops can be inlined.
|
|
bool isLegalToInline(Operation *, Region *, bool, IRMapping &) const final {
|
|
return true;
|
|
}
|
|
};
|
|
} // namespace
|
|
|
|
void arith::ArithDialect::initialize() {
|
|
addOperations<
|
|
#define GET_OP_LIST
|
|
#include "mlir/Dialect/Arith/IR/ArithOps.cpp.inc"
|
|
>();
|
|
addAttributes<
|
|
#define GET_ATTRDEF_LIST
|
|
#include "mlir/Dialect/Arith/IR/ArithOpsAttributes.cpp.inc"
|
|
>();
|
|
addInterfaces<ArithInlinerInterface>();
|
|
}
|
|
|
|
/// Materialize an integer or floating point constant.
|
|
Operation *arith::ArithDialect::materializeConstant(OpBuilder &builder,
|
|
Attribute value, Type type,
|
|
Location loc) {
|
|
if (auto poison = dyn_cast<ub::PoisonAttr>(value))
|
|
return builder.create<ub::PoisonOp>(loc, type, poison);
|
|
|
|
return ConstantOp::materialize(builder, value, type, loc);
|
|
}
|