ca21a65dd8
fixes #181901 This change detects when a HLSL Cbuffer matrix’s layout differs from its in-memory type and materializes a temporary with the non-padded matrix type. Matrix elements are copied explicitly from the padded buffer layout into the temporary, ensuring correct addressing and avoiding overlapping GEPs or incorrect vector flattening.
62 lines
2.4 KiB
C++
62 lines
2.4 KiB
C++
//===- HLSLBufferLayoutBuilder.h ------------------------------------------===//
|
|
//
|
|
// 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 "clang/AST/TypeBase.h"
|
|
#include "llvm/ADT/StringRef.h"
|
|
#include "llvm/IR/DerivedTypes.h"
|
|
|
|
namespace clang {
|
|
namespace CodeGen {
|
|
class CGHLSLOffsetInfo;
|
|
class CodeGenModule;
|
|
class CGHLSLOffsetInfo;
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Implementation of constant buffer layout common between DirectX and
|
|
// SPIR/SPIR-V.
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
class HLSLBufferLayoutBuilder {
|
|
private:
|
|
CodeGenModule &CGM;
|
|
|
|
/// Pads an array of elements to 16-byte cbuffer row boundaries.
|
|
/// This implements the common pattern of padding all-but-the-last element.
|
|
llvm::Type *padArrayElements(llvm::Type *EltTy, uint64_t Count);
|
|
|
|
public:
|
|
HLSLBufferLayoutBuilder(CodeGenModule &CGM) : CGM(CGM) {}
|
|
|
|
/// Lays out a struct type following HLSL buffer rules and considering any
|
|
/// explicit offset information. Previously created layout structs are cached
|
|
/// by CGHLSLRuntime.
|
|
///
|
|
/// The function iterates over all fields of the record type (including base
|
|
/// classes) and works out a padded llvm type to represent the buffer layout.
|
|
///
|
|
/// If a non-empty OffsetInfo is provided (ie, from `packoffset` annotations
|
|
/// in the source), any provided offsets offsets will be respected. If the
|
|
/// OffsetInfo is available but has empty entries, those will be layed out at
|
|
/// the end of the structure.
|
|
llvm::StructType *layOutStruct(const RecordType *StructType,
|
|
const CGHLSLOffsetInfo &OffsetInfo);
|
|
|
|
/// Lays out an array type following HLSL buffer rules.
|
|
llvm::Type *layOutArray(const ConstantArrayType *AT);
|
|
|
|
/// Lays out a matrix type following HLSL buffer rules.
|
|
llvm::Type *layOutMatrix(const ConstantMatrixType *MT);
|
|
|
|
/// Lays out a type following HLSL buffer rules. Arrays and structures will be
|
|
/// padded appropriately and nested objects will be converted as appropriate.
|
|
llvm::Type *layOutType(QualType Type);
|
|
};
|
|
|
|
} // namespace CodeGen
|
|
} // namespace clang
|