42 lines
1.1 KiB
C++
42 lines
1.1 KiB
C++
|
//===- 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]);
|
||
|
|
}
|
||
|
|
}
|