57 lines
1.8 KiB
C++
57 lines
1.8 KiB
C++
/*
|
|
* Copyright (C) 2021-2023 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <array>
|
|
#include <cstdint>
|
|
#include <optional>
|
|
#include <type_traits>
|
|
|
|
namespace NEO {
|
|
|
|
struct KernelDescriptor;
|
|
|
|
struct ImplicitArgs {
|
|
uint8_t structSize;
|
|
uint8_t structVersion;
|
|
uint8_t numWorkDim;
|
|
uint8_t simdWidth;
|
|
uint32_t localSizeX;
|
|
uint32_t localSizeY;
|
|
uint32_t localSizeZ;
|
|
uint64_t globalSizeX;
|
|
uint64_t globalSizeY;
|
|
uint64_t globalSizeZ;
|
|
uint64_t printfBufferPtr;
|
|
uint64_t globalOffsetX;
|
|
uint64_t globalOffsetY;
|
|
uint64_t globalOffsetZ;
|
|
uint64_t localIdTablePtr;
|
|
uint32_t groupCountX;
|
|
uint32_t groupCountY;
|
|
uint32_t groupCountZ;
|
|
uint32_t padding0;
|
|
uint64_t rtGlobalBufferPtr;
|
|
uint64_t assertBufferPtr;
|
|
uint8_t reserved[16];
|
|
};
|
|
|
|
static_assert((sizeof(ImplicitArgs) & 31) == 0, "Implicit args size need to be aligned to 32");
|
|
static_assert(sizeof(ImplicitArgs) == 128);
|
|
static_assert(std::is_pod<ImplicitArgs>::value);
|
|
|
|
inline constexpr const char *implicitArgsRelocationSymbolName = "__INTEL_PATCH_CROSS_THREAD_OFFSET_OFF_R0";
|
|
|
|
namespace ImplicitArgsHelper {
|
|
std::array<uint8_t, 3> getDimensionOrderForLocalIds(const uint8_t *workgroupDimensionsOrder, std::optional<std::pair<bool /* localIdsGeneratedByRuntime */, uint32_t /* walkOrderForHwGenerationOfLocalIds */>> hwGenerationOfLocalIdsParams);
|
|
uint32_t getGrfSize(uint32_t simd);
|
|
uint32_t getSizeForImplicitArgsPatching(const ImplicitArgs *pImplicitArgs, const KernelDescriptor &kernelDescriptor);
|
|
void *patchImplicitArgs(void *ptrToPatch, const ImplicitArgs &implicitArgs, const KernelDescriptor &kernelDescriptor, std::optional<std::pair<bool /* localIdsGeneratedByRuntime */, uint32_t /* walkOrderForHwGenerationOfLocalIds */>> hwGenerationOfLocalIdsParams);
|
|
} // namespace ImplicitArgsHelper
|
|
} // namespace NEO
|