llvm/mlir/unittests/Dialect/MemRef/InferShapeTest.cpp

//===- InferShapeTest.cpp - unit tests for shape inference ----------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

#include "mlir/Dialect/MemRef/IR/MemRef.h"
#include "mlir/IR/AffineMap.h"
#include "mlir/IR/Builders.h"
#include "mlir/IR/BuiltinTypes.h"
#include "gtest/gtest.h"

usingnamespacemlir;
usingnamespacemlir::memref;

// Source memref has identity layout.
TEST(InferShapeTest, inferRankReducedShapeIdentity) { … }

// Source memref has non-identity layout.
TEST(InferShapeTest, inferRankReducedShapeNonIdentity) { … }

TEST(InferShapeTest, inferRankReducedShapeToScalar) { … }