[all-commits] [llvm/llvm-project] b2505c: [MLIR] Allow ShapedTypeComponents with attribute i...

[Smit Hinsu via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: b2505ca2ece3174193f19599a6677b5d0257654a
      https://github.com/llvm/llvm-project/commit/b2505ca2ece3174193f19599a6677b5d0257654a
  Author: smit-hinsu <smittvhinsu at gmail.com>
  Date:   2022-12-07 (Wed, 07 Dec 2022)

  Changed paths:
    M mlir/lib/Interfaces/InferTypeOpInterface.cpp
    M mlir/test/lib/Dialect/Test/TestDialect.cpp
    M mlir/test/mlir-tblgen/return-types.mlir

  Log Message:
  -----------
  [MLIR] Allow ShapedTypeComponents with attribute in inferReturnTensorTypes

Originally, inferReturnTensorTypes didn't support shaped type components
containing an attribute just because there wasn't any motivating use-case.
Removing that limitation and using it to set the encoding attribute for
RankedTensorType.

Updated the existing test to set result attribute based on the first operand,
if available.

Signed-off-by: Smit Hinsu <smittvhinsu at gmail.com>

Differential Revision: https://reviews.llvm.org/D139271