142 lines
4.8 KiB
C++
142 lines
4.8 KiB
C++
/*
|
|
* Copyright (C) 2018-2021 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
#include "shared/source/helpers/hw_info.h"
|
|
#include "shared/source/kernel/kernel_descriptor.h"
|
|
#include "shared/source/utilities/arrayref.h"
|
|
#include "shared/source/utilities/const_stringref.h"
|
|
|
|
#include "opencl/source/program/heap_info.h"
|
|
|
|
#include <algorithm>
|
|
#include <array>
|
|
#include <cmath>
|
|
#include <cstdint>
|
|
#include <map>
|
|
#include <string>
|
|
#include <unordered_map>
|
|
#include <vector>
|
|
|
|
namespace gtpin {
|
|
typedef struct igc_info_s igc_info_t;
|
|
}
|
|
|
|
namespace NEO {
|
|
class BuiltinDispatchInfoBuilder;
|
|
class Device;
|
|
class Kernel;
|
|
struct KernelInfo;
|
|
class DispatchInfo;
|
|
struct KernelArgumentType;
|
|
class GraphicsAllocation;
|
|
class MemoryManager;
|
|
|
|
extern std::map<std::string, size_t> typeSizeMap;
|
|
|
|
static const float YTilingRatioValue = 1.3862943611198906188344642429164f;
|
|
|
|
struct WorkSizeInfo {
|
|
uint32_t maxWorkGroupSize;
|
|
uint32_t minWorkGroupSize;
|
|
bool hasBarriers;
|
|
uint32_t simdSize;
|
|
uint32_t slmTotalSize;
|
|
GFXCORE_FAMILY coreFamily;
|
|
uint32_t numThreadsPerSubSlice;
|
|
uint32_t localMemSize;
|
|
bool imgUsed = false;
|
|
bool yTiledSurfaces = false;
|
|
bool useRatio = false;
|
|
bool useStrictRatio = false;
|
|
float targetRatio = 0;
|
|
|
|
WorkSizeInfo(uint32_t maxWorkGroupSize, bool hasBarriers, uint32_t simdSize, uint32_t slmTotalSize, GFXCORE_FAMILY coreFamily, uint32_t numThreadsPerSubSlice, uint32_t localMemSize, bool imgUsed, bool yTiledSurface);
|
|
WorkSizeInfo(const DispatchInfo &dispatchInfo);
|
|
void setIfUseImg(const KernelInfo &kernelInfo);
|
|
void setMinWorkGroupSize();
|
|
void checkRatio(const size_t workItems[3]);
|
|
};
|
|
|
|
struct DeviceInfoKernelPayloadConstants {
|
|
void *slmWindow = nullptr;
|
|
uint32_t slmWindowSize = 0U;
|
|
uint32_t computeUnitsUsedForScratch = 0U;
|
|
uint32_t maxWorkGroupSize = 0U;
|
|
};
|
|
|
|
struct KernelInfo {
|
|
public:
|
|
KernelInfo() = default;
|
|
KernelInfo(const KernelInfo &) = delete;
|
|
KernelInfo &operator=(const KernelInfo &) = delete;
|
|
~KernelInfo();
|
|
|
|
GraphicsAllocation *getGraphicsAllocation() const { return this->kernelAllocation; }
|
|
|
|
const ArgDescriptor &getArgDescriptorAt(uint32_t index) const {
|
|
DEBUG_BREAK_IF(index >= kernelDescriptor.payloadMappings.explicitArgs.size());
|
|
return kernelDescriptor.payloadMappings.explicitArgs[index];
|
|
}
|
|
const StackVec<ArgDescriptor, 16> &getExplicitArgs() const {
|
|
return kernelDescriptor.payloadMappings.explicitArgs;
|
|
}
|
|
const ArgTypeMetadataExtended &getExtendedMetadata(uint32_t index) const {
|
|
DEBUG_BREAK_IF(index >= kernelDescriptor.explicitArgsExtendedMetadata.size());
|
|
return kernelDescriptor.explicitArgsExtendedMetadata[index];
|
|
}
|
|
size_t getSamplerStateArrayCount() const;
|
|
size_t getSamplerStateArraySize(const HardwareInfo &hwInfo) const;
|
|
size_t getBorderColorStateSize() const;
|
|
size_t getBorderColorOffset() const;
|
|
unsigned int getMaxSimdSize() const {
|
|
return kernelDescriptor.kernelAttributes.simdSize;
|
|
}
|
|
bool hasDeviceEnqueue() const {
|
|
return kernelDescriptor.kernelAttributes.flags.usesDeviceSideEnqueue;
|
|
}
|
|
bool requiresSubgroupIndependentForwardProgress() const {
|
|
return kernelDescriptor.kernelAttributes.flags.requiresSubgroupIndependentForwardProgress;
|
|
}
|
|
size_t getMaxRequiredWorkGroupSize(size_t maxWorkGroupSize) const {
|
|
auto requiredWorkGroupSizeX = kernelDescriptor.kernelAttributes.requiredWorkgroupSize[0];
|
|
auto requiredWorkGroupSizeY = kernelDescriptor.kernelAttributes.requiredWorkgroupSize[1];
|
|
auto requiredWorkGroupSizeZ = kernelDescriptor.kernelAttributes.requiredWorkgroupSize[2];
|
|
size_t maxRequiredWorkGroupSize = requiredWorkGroupSizeX * requiredWorkGroupSizeY * requiredWorkGroupSizeZ;
|
|
if ((maxRequiredWorkGroupSize == 0) || (maxRequiredWorkGroupSize > maxWorkGroupSize)) {
|
|
maxRequiredWorkGroupSize = maxWorkGroupSize;
|
|
}
|
|
return maxRequiredWorkGroupSize;
|
|
}
|
|
|
|
uint32_t getConstantBufferSize() const;
|
|
int32_t getArgNumByName(const char *name) const;
|
|
|
|
bool createKernelAllocation(const Device &device, bool internalIsa);
|
|
void apply(const DeviceInfoKernelPayloadConstants &constants);
|
|
|
|
HeapInfo heapInfo = {};
|
|
std::vector<std::pair<uint32_t, uint32_t>> childrenKernelsIdOffset;
|
|
char *crossThreadData = nullptr;
|
|
const BuiltinDispatchInfoBuilder *builtinDispatchBuilder = nullptr;
|
|
uint32_t systemKernelOffset = 0;
|
|
uint64_t kernelId = 0;
|
|
bool hasIndirectStatelessAccess = false;
|
|
bool isKernelHeapSubstituted = false;
|
|
GraphicsAllocation *kernelAllocation = nullptr;
|
|
DebugData debugData;
|
|
bool computeMode = false;
|
|
const gtpin::igc_info_t *igcInfoForGtpin = nullptr;
|
|
|
|
uint64_t shaderHashCode;
|
|
KernelDescriptor kernelDescriptor;
|
|
};
|
|
|
|
std::string concatenateKernelNames(ArrayRef<KernelInfo *> kernelInfos);
|
|
|
|
} // namespace NEO
|