summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--mlir/include/mlir/Dialect/Linalg/IR/LinalgStructuredOps.td4
-rw-r--r--mlir/include/mlir/Dialect/Linalg/IR/LinalgTraits.h3
2 files changed, 4 insertions, 3 deletions
diff --git a/mlir/include/mlir/Dialect/Linalg/IR/LinalgStructuredOps.td b/mlir/include/mlir/Dialect/Linalg/IR/LinalgStructuredOps.td
index dd9e09b8eae..ed09272055b 100644
--- a/mlir/include/mlir/Dialect/Linalg/IR/LinalgStructuredOps.td
+++ b/mlir/include/mlir/Dialect/Linalg/IR/LinalgStructuredOps.td
@@ -29,7 +29,7 @@ class NInputs<int args_in> :
class NOutputs<int args_out> :
NativeOpTrait<"linalg::NOutputs<" # !cast<string>(args_out) # ">::Impl"> {}
-def ViewTraits : NativeOpTrait<"linalg::ViewTraits">;
+def StructuredOpTraits : NativeOpTrait<"linalg::StructuredOpTraits">;
// The linalg 'LinalgStructuredInterface' provides access to the 'LinalgOp'
// interface.
@@ -147,7 +147,7 @@ def LinalgStructuredInterface : OpInterface<"LinalgOp"> {
// depending on the specific Linalg op.
class LinalgStructuredBase_Op<string mnemonic, list<OpTrait> props>
: Op<Linalg_Dialect, mnemonic,
- !listconcat(props, [ViewTraits, LinalgStructuredInterface])> {
+ !listconcat(props, [StructuredOpTraits, LinalgStructuredInterface])> {
let parser = [{ return parseLinalgStructuredOp(parser, result); }];
let printer = [{ printLinalgStructuredOp(p, *this); }];
}
diff --git a/mlir/include/mlir/Dialect/Linalg/IR/LinalgTraits.h b/mlir/include/mlir/Dialect/Linalg/IR/LinalgTraits.h
index e0d651806d3..0706f1fd363 100644
--- a/mlir/include/mlir/Dialect/Linalg/IR/LinalgTraits.h
+++ b/mlir/include/mlir/Dialect/Linalg/IR/LinalgTraits.h
@@ -55,7 +55,8 @@ public:
/// class DotOp : public Op<DotOp, OpTrait::ViewTrait> {
///
template <typename ConcreteType>
-class ViewTraits : public OpTrait::TraitBase<ConcreteType, ViewTraits> {
+class StructuredOpTraits
+ : public OpTrait::TraitBase<ConcreteType, StructuredOpTraits> {
private:
/// Return the number of input views. For internal use only.
unsigned nInputs() {
OpenPOWER on IntegriCloud