154 lines
7.1 KiB
C++
154 lines
7.1 KiB
C++
//===- llvm/unittests/Target/DirectX/CBufferDataLayoutTests.cpp -----------===//
|
|
//
|
|
// 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 "CBufferDataLayout.h"
|
|
#include "llvm/IR/DataLayout.h"
|
|
#include "llvm/IR/Instructions.h"
|
|
#include "llvm/IR/LLVMContext.h"
|
|
#include "llvm/IR/Type.h"
|
|
|
|
#include "gmock/gmock.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
using ::testing::Contains;
|
|
using ::testing::Pair;
|
|
|
|
using namespace llvm;
|
|
using namespace llvm::dxil;
|
|
|
|
void checkLegacyLayout(CBufferDataLayout &CBDL, Type *T16, Type *T32,
|
|
Type *T64) {
|
|
// Basic types.
|
|
EXPECT_EQ(2ULL, CBDL.getTypeAllocSizeInBytes(T16).getFixedValue());
|
|
EXPECT_EQ(4ULL, CBDL.getTypeAllocSizeInBytes(T32).getFixedValue());
|
|
EXPECT_EQ(8ULL, CBDL.getTypeAllocSizeInBytes(T64).getFixedValue());
|
|
// Vector types.
|
|
Type *T16V2 = FixedVectorType::get(T16, 2);
|
|
Type *T32V2 = FixedVectorType::get(T32, 2);
|
|
Type *T64V2 = FixedVectorType::get(T64, 2);
|
|
Type *T16V3 = FixedVectorType::get(T16, 3);
|
|
Type *T32V3 = FixedVectorType::get(T32, 3);
|
|
Type *T64V3 = FixedVectorType::get(T64, 3);
|
|
Type *T16V4 = FixedVectorType::get(T16, 4);
|
|
Type *T32V4 = FixedVectorType::get(T32, 4);
|
|
Type *T64V4 = FixedVectorType::get(T64, 4);
|
|
EXPECT_EQ(4ULL, CBDL.getTypeAllocSizeInBytes(T16V2).getFixedValue());
|
|
EXPECT_EQ(8ULL, CBDL.getTypeAllocSizeInBytes(T32V2).getFixedValue());
|
|
EXPECT_EQ(16ULL, CBDL.getTypeAllocSizeInBytes(T64V2).getFixedValue());
|
|
EXPECT_EQ(6ULL, CBDL.getTypeAllocSizeInBytes(T16V3).getFixedValue());
|
|
EXPECT_EQ(12ULL, CBDL.getTypeAllocSizeInBytes(T32V3).getFixedValue());
|
|
EXPECT_EQ(24ULL, CBDL.getTypeAllocSizeInBytes(T64V3).getFixedValue());
|
|
EXPECT_EQ(8ULL, CBDL.getTypeAllocSizeInBytes(T16V4).getFixedValue());
|
|
EXPECT_EQ(16ULL, CBDL.getTypeAllocSizeInBytes(T32V4).getFixedValue());
|
|
EXPECT_EQ(32ULL, CBDL.getTypeAllocSizeInBytes(T64V4).getFixedValue());
|
|
|
|
// Array types.
|
|
|
|
ArrayType *T16A3 = ArrayType::get(T16, 3);
|
|
ArrayType *T32A3 = ArrayType::get(T32, 3);
|
|
ArrayType *T64A3 = ArrayType::get(T64, 3);
|
|
|
|
EXPECT_EQ(34ULL, CBDL.getTypeAllocSizeInBytes(T16A3).getFixedValue());
|
|
EXPECT_EQ(36ULL, CBDL.getTypeAllocSizeInBytes(T32A3).getFixedValue());
|
|
EXPECT_EQ(40ULL, CBDL.getTypeAllocSizeInBytes(T64A3).getFixedValue());
|
|
|
|
ArrayType *T16V3A3 = ArrayType::get(T16V3, 3);
|
|
ArrayType *T32V3A3 = ArrayType::get(T32V3, 3);
|
|
ArrayType *T64V3A3 = ArrayType::get(T64V3, 3);
|
|
|
|
EXPECT_EQ(38ULL, CBDL.getTypeAllocSizeInBytes(T16V3A3).getFixedValue());
|
|
EXPECT_EQ(44ULL, CBDL.getTypeAllocSizeInBytes(T32V3A3).getFixedValue());
|
|
EXPECT_EQ(88ULL, CBDL.getTypeAllocSizeInBytes(T64V3A3).getFixedValue());
|
|
|
|
ArrayType *T16V3A3A3 = ArrayType::get(T16V3A3, 3);
|
|
ArrayType *T32V3A3A3 = ArrayType::get(T32V3A3, 3);
|
|
ArrayType *T64V3A3A3 = ArrayType::get(T64V3A3, 3);
|
|
|
|
EXPECT_EQ((48 * 2 + 38ULL),
|
|
CBDL.getTypeAllocSizeInBytes(T16V3A3A3).getFixedValue());
|
|
EXPECT_EQ((48 * 2 + 44ULL),
|
|
CBDL.getTypeAllocSizeInBytes(T32V3A3A3).getFixedValue());
|
|
EXPECT_EQ((96 * 2 + 88ULL),
|
|
CBDL.getTypeAllocSizeInBytes(T64V3A3A3).getFixedValue());
|
|
|
|
// Struct types.
|
|
StructType *BasicMix0 = StructType::get(T16, T32, T64);
|
|
StructType *BasicMix1 = StructType::get(T16, T64, T32);
|
|
StructType *BasicMix2 = StructType::get(T32, T64, T16);
|
|
StructType *BasicMix3 = StructType::get(T32, T16, T64);
|
|
StructType *BasicMix4 = StructType::get(T64, T16, T32);
|
|
StructType *BasicMix5 = StructType::get(T64, T32, T16);
|
|
|
|
EXPECT_EQ(16ULL, CBDL.getTypeAllocSizeInBytes(BasicMix0).getFixedValue());
|
|
EXPECT_EQ(20ULL, CBDL.getTypeAllocSizeInBytes(BasicMix1).getFixedValue());
|
|
EXPECT_EQ(18ULL, CBDL.getTypeAllocSizeInBytes(BasicMix2).getFixedValue());
|
|
EXPECT_EQ(16ULL, CBDL.getTypeAllocSizeInBytes(BasicMix3).getFixedValue());
|
|
EXPECT_EQ(16ULL, CBDL.getTypeAllocSizeInBytes(BasicMix4).getFixedValue());
|
|
EXPECT_EQ(14ULL, CBDL.getTypeAllocSizeInBytes(BasicMix5).getFixedValue());
|
|
|
|
StructType *VecMix0 = StructType::get(T16V3, T16, T32, T64V2);
|
|
StructType *VecMix1 = StructType::get(T16V3, T32, T64V2, T16);
|
|
StructType *VecMix2 = StructType::get(T16V3, T64, T32V2, T16);
|
|
StructType *VecMix3 = StructType::get(T32V3, T64, T16V2, T32);
|
|
StructType *VecMix4 = StructType::get(T32V3, T16, T16V2, T64);
|
|
StructType *VecMix5 = StructType::get(T32V3, T64V3, T16V2, T64);
|
|
|
|
EXPECT_EQ(32ULL, CBDL.getTypeAllocSizeInBytes(VecMix0).getFixedValue());
|
|
EXPECT_EQ(34ULL, CBDL.getTypeAllocSizeInBytes(VecMix1).getFixedValue());
|
|
EXPECT_EQ(26ULL, CBDL.getTypeAllocSizeInBytes(VecMix2).getFixedValue());
|
|
EXPECT_EQ(32ULL, CBDL.getTypeAllocSizeInBytes(VecMix3).getFixedValue());
|
|
EXPECT_EQ(32ULL, CBDL.getTypeAllocSizeInBytes(VecMix4).getFixedValue());
|
|
EXPECT_EQ(56ULL, CBDL.getTypeAllocSizeInBytes(VecMix5).getFixedValue());
|
|
|
|
StructType *ArrayMix0 = StructType::get(T16A3, T16, T32, T64A3);
|
|
StructType *ArrayMix1 = StructType::get(T32A3, T16, T32, T16A3);
|
|
StructType *ArrayMix2 = StructType::get(T16A3, T32, T64, T32A3);
|
|
StructType *ArrayMix3 = StructType::get(T32A3, T32, T64, T16A3);
|
|
StructType *ArrayMix4 = StructType::get(T16A3, T64, T16, T64A3);
|
|
StructType *ArrayMix5 = StructType::get(T32A3, T64, T16, T32A3);
|
|
|
|
EXPECT_EQ(88ULL, CBDL.getTypeAllocSizeInBytes(ArrayMix0).getFixedValue());
|
|
EXPECT_EQ(82ULL, CBDL.getTypeAllocSizeInBytes(ArrayMix1).getFixedValue());
|
|
EXPECT_EQ(84ULL, CBDL.getTypeAllocSizeInBytes(ArrayMix2).getFixedValue());
|
|
EXPECT_EQ(82ULL, CBDL.getTypeAllocSizeInBytes(ArrayMix3).getFixedValue());
|
|
EXPECT_EQ(104ULL, CBDL.getTypeAllocSizeInBytes(ArrayMix4).getFixedValue());
|
|
EXPECT_EQ(100ULL, CBDL.getTypeAllocSizeInBytes(ArrayMix5).getFixedValue());
|
|
|
|
StructType *StructMix0 = StructType::get(T16A3, T16, T32, ArrayMix0);
|
|
StructType *StructMix1 = StructType::get(StructMix0, T16, T32, ArrayMix1);
|
|
ArrayType *StructArray0 = ArrayType::get(StructMix1, 3);
|
|
StructType *StructMix2 = StructType::get(StructArray0, T64);
|
|
ArrayType *StructArray1 = ArrayType::get(StructMix2, 3);
|
|
StructType *StructMix3 = StructType::get(StructArray1, T32);
|
|
EXPECT_EQ(136ULL, CBDL.getTypeAllocSizeInBytes(StructMix0).getFixedValue());
|
|
EXPECT_EQ(226ULL, CBDL.getTypeAllocSizeInBytes(StructMix1).getFixedValue());
|
|
EXPECT_EQ(706ULL, CBDL.getTypeAllocSizeInBytes(StructArray0).getFixedValue());
|
|
EXPECT_EQ(720ULL, CBDL.getTypeAllocSizeInBytes(StructMix2).getFixedValue());
|
|
EXPECT_EQ(2160ULL,
|
|
CBDL.getTypeAllocSizeInBytes(StructArray1).getFixedValue());
|
|
EXPECT_EQ(2164ULL, CBDL.getTypeAllocSizeInBytes(StructMix3).getFixedValue());
|
|
}
|
|
|
|
TEST(CBufferDataLayout, LegacyLayout) {
|
|
LLVMContext Context;
|
|
llvm::DataLayout DL("e-m:e-p:32:32-i1:32-i8:8-i16:16-i32:32-i64:64-f16:16-"
|
|
"f32:32-f64:64-n8:16:32:64");
|
|
|
|
CBufferDataLayout CBDL(DL, true);
|
|
|
|
Type *F16 = Type::getHalfTy(Context);
|
|
Type *F32 = Type::getFloatTy(Context);
|
|
Type *F64 = Type::getDoubleTy(Context);
|
|
|
|
Type *I16 = Type::getInt16Ty(Context);
|
|
Type *I32 = Type::getInt32Ty(Context);
|
|
Type *I64 = Type::getInt64Ty(Context);
|
|
|
|
checkLegacyLayout(CBDL, F16, F32, F64);
|
|
checkLegacyLayout(CBDL, I16, I32, I64);
|
|
}
|