de467afe18
This patch introduces a bunch of builder functions to create function calls to runtime ragged arrays functions. This patch is part of the upstreaming effort from fir-dev branch. Reviewed By: kiranchandramohan Differential Revision: https://reviews.llvm.org/D114535 Co-authored-by: Eric Schweitz <eschweitz@nvidia.com>
37 lines
1.6 KiB
C++
37 lines
1.6 KiB
C++
//===- RaggedTest.cpp -- Ragged array runtime function builder unit tests -===//
|
|
//
|
|
// 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 "flang/Optimizer/Builder/Runtime/Ragged.h"
|
|
#include "RuntimeCallTestBase.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
TEST_F(RuntimeCallTest, genRaggedArrayAllocateTest) {
|
|
auto loc = firBuilder->getUnknownLoc();
|
|
mlir::TupleType headerTy =
|
|
fir::factory::getRaggedArrayHeaderType(*firBuilder);
|
|
mlir::Value header = firBuilder->create<fir::UndefOp>(loc, headerTy);
|
|
mlir::Value eleSize = firBuilder->createIntegerConstant(loc, i32Ty, 1);
|
|
mlir::Value extent = firBuilder->createIntegerConstant(loc, i32Ty, 1);
|
|
// Use a dummy header just to test the correctness of the generated call.
|
|
fir::runtime::genRaggedArrayAllocate(
|
|
loc, *firBuilder, header, false, eleSize, {extent});
|
|
checkCallOpFromResultBox(
|
|
eleSize, "_FortranARaggedArrayAllocate", 5, /*addLocArgs=*/false);
|
|
}
|
|
|
|
TEST_F(RuntimeCallTest, genRaggedArrayDeallocateTest) {
|
|
auto loc = firBuilder->getUnknownLoc();
|
|
mlir::TupleType headerTy =
|
|
fir::factory::getRaggedArrayHeaderType(*firBuilder);
|
|
// Use a dummy header just to test the correctness of the generated call.
|
|
mlir::Value header = firBuilder->create<fir::UndefOp>(loc, headerTy);
|
|
fir::runtime::genRaggedArrayDeallocate(loc, *firBuilder, header);
|
|
checkCallOpFromResultBox(
|
|
header, "_FortranARaggedArrayDeallocate", 1, /*addLocArgs=*/false);
|
|
}
|