diff options
-rw-r--r-- | mlir/include/mlir/Dialect/Linalg/IR/LinalgStructuredOps.td | 4 | ||||
-rw-r--r-- | mlir/include/mlir/Dialect/Linalg/IR/LinalgTraits.h | 3 |
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() { |