compute-runtime/opencl/source/helpers/per_thread_data.h

54 lines
1.4 KiB
C++

/*
* Copyright (C) 2017-2020 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
*/
#pragma once
#include "opencl/source/command_queue/local_id_gen.h"
#include "patch_shared.h"
#include <array>
#include <cstddef>
#include <cstdint>
namespace NEO {
class LinearStream;
struct PerThreadDataHelper {
static inline uint32_t getLocalIdSizePerThread(
uint32_t simd,
uint32_t grfSize,
uint32_t numChannels) {
return getPerThreadSizeLocalIDs(simd, grfSize, numChannels);
}
static inline size_t getPerThreadDataSizeTotal(
uint32_t simd,
uint32_t grfSize,
uint32_t numChannels,
size_t localWorkSize) {
return getThreadsPerWG(simd, localWorkSize) * getLocalIdSizePerThread(simd, grfSize, numChannels);
}
static size_t sendPerThreadData(
LinearStream &indirectHeap,
uint32_t simd,
uint32_t grfSize,
uint32_t numChannels,
const size_t localWorkSizes[3],
const std::array<uint8_t, 3> &workgroupWalkOrder,
bool hasKernelOnlyImages);
static inline uint32_t getNumLocalIdChannels(const iOpenCL::SPatchThreadPayload &threadPayload) {
return threadPayload.LocalIDXPresent +
threadPayload.LocalIDYPresent +
threadPayload.LocalIDZPresent;
}
static uint32_t getThreadPayloadSize(const iOpenCL::SPatchThreadPayload &threadPayload, uint32_t simd, uint32_t grfSize);
};
} // namespace NEO