Redefines the amd_kernel_code_t struct with MCExprs for members that would be derived from SIProgramInfo MCExpr members.
91 lines
3.2 KiB
C++
91 lines
3.2 KiB
C++
//===- AMDGPUKernelCodeTUtils.h - helpers for amd_kernel_code_t -*- C++ -*-===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
/// \file AMDKernelCodeTUtils.h
|
|
/// MC layer struct for AMDGPUMCKernelCodeT, provides MCExpr functionality where
|
|
/// required.
|
|
///
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_LIB_TARGET_AMDGPU_MCTARGETDESC_AMDGPUMCKERNELCODET_H
|
|
#define LLVM_LIB_TARGET_AMDGPU_MCTARGETDESC_AMDGPUMCKERNELCODET_H
|
|
|
|
#include "AMDKernelCodeT.h"
|
|
#include "llvm/ADT/ArrayRef.h"
|
|
#include "llvm/ADT/StringRef.h"
|
|
|
|
namespace llvm {
|
|
class MCAsmParser;
|
|
class MCContext;
|
|
class MCExpr;
|
|
class MCStreamer;
|
|
class MCSubtargetInfo;
|
|
class raw_ostream;
|
|
namespace AMDGPU {
|
|
|
|
struct AMDGPUMCKernelCodeT {
|
|
AMDGPUMCKernelCodeT() = default;
|
|
|
|
// Names of most (if not all) members should match the ones used for table
|
|
// driven (array) generation in AMDKernelCodeTInfo.h.
|
|
uint32_t amd_kernel_code_version_major = 0;
|
|
uint32_t amd_kernel_code_version_minor = 0;
|
|
uint16_t amd_machine_kind = 0;
|
|
uint16_t amd_machine_version_major = 0;
|
|
uint16_t amd_machine_version_minor = 0;
|
|
uint16_t amd_machine_version_stepping = 0;
|
|
int64_t kernel_code_entry_byte_offset = 0;
|
|
int64_t kernel_code_prefetch_byte_offset = 0;
|
|
uint64_t kernel_code_prefetch_byte_size = 0;
|
|
uint64_t reserved0 = 0;
|
|
uint64_t compute_pgm_resource_registers = 0;
|
|
uint32_t code_properties = 0;
|
|
uint32_t workgroup_group_segment_byte_size = 0;
|
|
uint32_t gds_segment_byte_size = 0;
|
|
uint64_t kernarg_segment_byte_size = 0;
|
|
uint32_t workgroup_fbarrier_count = 0;
|
|
uint16_t reserved_vgpr_first = 0;
|
|
uint16_t reserved_vgpr_count = 0;
|
|
uint16_t reserved_sgpr_first = 0;
|
|
uint16_t reserved_sgpr_count = 0;
|
|
uint16_t debug_wavefront_private_segment_offset_sgpr = 0;
|
|
uint16_t debug_private_segment_buffer_sgpr = 0;
|
|
uint8_t kernarg_segment_alignment = 0;
|
|
uint8_t group_segment_alignment = 0;
|
|
uint8_t private_segment_alignment = 0;
|
|
uint8_t wavefront_size = 0;
|
|
int32_t call_convention = 0;
|
|
uint8_t reserved3[12] = {0};
|
|
uint64_t runtime_loader_kernel_symbol = 0;
|
|
uint64_t control_directives[16] = {0};
|
|
|
|
const MCExpr *compute_pgm_resource1_registers = nullptr;
|
|
const MCExpr *compute_pgm_resource2_registers = nullptr;
|
|
|
|
const MCExpr *is_dynamic_callstack = nullptr;
|
|
const MCExpr *wavefront_sgpr_count = nullptr;
|
|
const MCExpr *workitem_vgpr_count = nullptr;
|
|
const MCExpr *workitem_private_segment_byte_size = nullptr;
|
|
|
|
void initDefault(const MCSubtargetInfo *STI, MCContext &Ctx,
|
|
bool InitMCExpr = true);
|
|
void validate(const MCSubtargetInfo *STI, MCContext &Ctx);
|
|
|
|
const MCExpr *&getMCExprForIndex(int Index);
|
|
|
|
bool ParseKernelCodeT(StringRef ID, MCAsmParser &MCParser, raw_ostream &Err);
|
|
void EmitKernelCodeT(raw_ostream &OS, MCContext &Ctx);
|
|
void EmitKernelCodeT(MCStreamer &OS, MCContext &Ctx);
|
|
};
|
|
|
|
} // end namespace AMDGPU
|
|
} // end namespace llvm
|
|
|
|
#endif // LLVM_LIB_TARGET_AMDGPU_MCTARGETDESC_AMDGPUMCKERNELCODET_H
|