forked from OSchip/llvm-project
29 lines
828 B
TableGen
29 lines
828 B
TableGen
// RUN: mlir-tblgen -gen-reference-implementations -I %S/../../include %s | FileCheck %s
|
|
|
|
#ifdef OP_BASE
|
|
#else
|
|
include "mlir/IR/OpBase.td"
|
|
#endif // OP_BASE
|
|
|
|
def X_Dialect : Dialect {
|
|
let name = "x";
|
|
}
|
|
class X_Op<string mnemonic, list<OpTrait> traits = []> :
|
|
Op<X_Dialect, mnemonic, traits>;
|
|
|
|
def X_AddOp : X_Op<"add">,
|
|
Arguments<(ins AnyTensor:$A, AnyTensor:$B)>,
|
|
Results<(outs AnyTensor: $C)> {
|
|
// TODO: extract referenceImplementation to Op.
|
|
code referenceImplementation = [{
|
|
auto ivs = IndexHandle::makeIndexHandles(view_A.rank());
|
|
auto pivs = IndexHandle::makeIndexHandlePointers(ivs);
|
|
IndexedValue A(arg_A), B(arg_B), C(arg_C);
|
|
LoopNestBuilder(pivs, view_A.getLbs(), view_A.getUbs(), view_A.getSteps())({
|
|
C(ivs) = A(ivs) + B(ivs)
|
|
});
|
|
}];
|
|
}
|
|
|
|
// CHECK: printRefImplementation
|