forked from OSchip/llvm-project
95 lines
3.7 KiB
TableGen
95 lines
3.7 KiB
TableGen
// RUN: mlir-tblgen -gen-op-defs -I %S/../../include %s | FileCheck %s
|
|
|
|
include "mlir/IR/OpBase.td"
|
|
|
|
def Test_Dialect : Dialect {
|
|
let name = "test";
|
|
}
|
|
class NS_Op<string mnemonic, list<OpTrait> traits> :
|
|
Op<Test_Dialect, mnemonic, traits>;
|
|
|
|
def I32OrF32 : Type<CPred<"$_self.isInteger(32) || $_self.isF32()">,
|
|
"32-bit integer or floating-point type">;
|
|
|
|
def OpA : NS_Op<"op_for_CPred_containing_multiple_same_placeholder", []> {
|
|
let arguments = (ins I32OrF32:$x);
|
|
}
|
|
|
|
// CHECK-LABEL: OpA::verify
|
|
// CHECK: for (Value v : getODSOperands(0)) {
|
|
// CHECK: if (!((v.getType().isInteger(32) || v.getType().isF32())))
|
|
|
|
def OpB : NS_Op<"op_for_And_PredOpTrait", [
|
|
PredOpTrait<"both first and second holds",
|
|
And<[CPred<"first">, CPred<"second">]>>]> {
|
|
}
|
|
|
|
// CHECK-LABEL: OpB::verify
|
|
// CHECK: if (!(((first)) && ((second))))
|
|
|
|
def OpF : NS_Op<"op_for_int_min_val", []> {
|
|
let arguments = (ins Confined<I32Attr, [IntMinValue<10>]>:$attr);
|
|
}
|
|
|
|
// CHECK-LABEL: OpF::verify()
|
|
// CHECK: (tblgen_attr.cast<IntegerAttr>().getInt() >= 10)
|
|
// CHECK-SAME: return emitOpError("attribute 'attr' failed to satisfy constraint: 32-bit signless integer attribute whose minimum value is 10");
|
|
|
|
def OpFX : NS_Op<"op_for_int_max_val", []> {
|
|
let arguments = (ins Confined<I32Attr, [IntMaxValue<10>]>:$attr);
|
|
}
|
|
|
|
// CHECK-LABEL: OpFX::verify()
|
|
// CHECK: (tblgen_attr.cast<IntegerAttr>().getInt() <= 10)
|
|
// CHECK-SAME: return emitOpError("attribute 'attr' failed to satisfy constraint: 32-bit signless integer attribute whose maximum value is 10");
|
|
|
|
def OpG : NS_Op<"op_for_arr_min_count", []> {
|
|
let arguments = (ins Confined<ArrayAttr, [ArrayMinCount<8>]>:$attr);
|
|
}
|
|
|
|
// CHECK-LABEL: OpG::verify()
|
|
// CHECK: (tblgen_attr.cast<ArrayAttr>().size() >= 8)
|
|
// CHECK-SAME: return emitOpError("attribute 'attr' failed to satisfy constraint: array attribute with at least 8 elements");
|
|
|
|
def OpH : NS_Op<"op_for_arr_value_at_index", []> {
|
|
let arguments = (ins Confined<ArrayAttr, [IntArrayNthElemEq<0, 8>]>:$attr);
|
|
}
|
|
|
|
// CHECK-LABEL: OpH::verify()
|
|
// CHECK: (((tblgen_attr.cast<ArrayAttr>().size() > 0)) && ((tblgen_attr.cast<ArrayAttr>()[0].cast<IntegerAttr>().getInt() == 8)))))
|
|
// CHECK-SAME: return emitOpError("attribute 'attr' failed to satisfy constraint: array attribute whose 0-th element must be 8");
|
|
|
|
def OpI: NS_Op<"op_for_arr_min_value_at_index", []> {
|
|
let arguments = (ins Confined<ArrayAttr, [IntArrayNthElemMinValue<0, 8>]>:$attr);
|
|
}
|
|
|
|
// CHECK-LABEL: OpI::verify()
|
|
// CHECK: (((tblgen_attr.cast<ArrayAttr>().size() > 0)) && ((tblgen_attr.cast<ArrayAttr>()[0].cast<IntegerAttr>().getInt() >= 8)))))
|
|
// CHECK-SAME: return emitOpError("attribute 'attr' failed to satisfy constraint: array attribute whose 0-th element must be at least 8");
|
|
|
|
def OpJ: NS_Op<"op_for_TCopVTEtAreSameAt", [
|
|
PredOpTrait<"operands indexed at 0, 2, 3 should all have "
|
|
"the same type", TCopVTEtAreSameAt<[0, 2, 3]>>]> {
|
|
let arguments = (ins
|
|
AnyTensor:$a,
|
|
AnyTensor:$b,
|
|
AnyTensor:$c,
|
|
AnyTensor:$d,
|
|
AnyTensor:$e
|
|
);
|
|
}
|
|
|
|
// CHECK-LABEL: OpJ::verify()
|
|
// CHECK: llvm::is_splat(mlir::functional::map(
|
|
// CHECK-SAME: [this](unsigned i) { return getElementTypeOrSelf(this->getOperand(i)); },
|
|
// CHECK-SAME: llvm::ArrayRef<unsigned>({0, 2, 3})))
|
|
// CHECK: return emitOpError("failed to verify that operands indexed at 0, 2, 3 should all have the same type");
|
|
|
|
def OpK : NS_Op<"op_for_AnyTensorOf", []> {
|
|
let arguments = (ins TensorOf<[F32, I32]>:$x);
|
|
}
|
|
|
|
// CHECK-LABEL: OpK::verify
|
|
// CHECK: for (Value v : getODSOperands(0)) {
|
|
// CHECK: if (!(((v.getType().isa<TensorType>())) && (((v.getType().cast<ShapedType>().getElementType().isF32())) || ((v.getType().cast<ShapedType>().getElementType().isSignlessInteger(32))))))
|