TheRock/compiler/amd-llvm/mlir/unittests/ExecutionEngine/StridedMemRef.cpp

//===- StridedMemRef.cpp ----------------------------------------*- C++ -*-===//
//
// This file is licensed 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/ExecutionEngine/CRunnerUtils.h"
#include "llvm/ADT/SmallVector.h"

#include "gmock/gmock.h"

using namespace ::mlir;
using namespace ::testing;

TEST(StridedMemRef, rankOneWithOffset) {
  std::array<int, 15> data;

  for (size_t i = 0; i < data.size(); ++i) {
    data[i] = i;
  }

  StridedMemRefType<int, 1> memRefA;
  memRefA.basePtr = data.data();
  memRefA.data = data.data();
  memRefA.offset = 0;
  memRefA.sizes[0] = 10;
  memRefA.strides[0] = 1;

  StridedMemRefType<int, 1> memRefB = memRefA;
  memRefB.offset = 5;

  llvm::SmallVector<int, 10> valuesA(memRefA.begin(), memRefA.end());
  llvm::SmallVector<int, 10> valuesB(memRefB.begin(), memRefB.end());

  for (int64_t i = 0; i < 10; ++i) {
    EXPECT_EQ(valuesA[i], i);
    EXPECT_EQ(valuesA[i] + 5, valuesB[i]);
  }
}