Home
last modified time | relevance | path

Searched refs:workgroupSize (Results 1 – 1 of 1) sorted by relevance

/packages/modules/NeuralNetworks/runtime/test/
DTestGpuNnapi.cpp147 uint32_t workgroupSize; member
163 uint32_t workgroupSize = std::min(workgroupInvocationsX, workgroupInvocationsY); in chooseDispatchSize() local
166 workgroupSize = std::min<uint32_t>(workgroupSize, limits.maxComputeWorkGroupSize[0]); in chooseDispatchSize()
167 workgroupSize = std::min<uint32_t>(workgroupSize, limits.maxComputeWorkGroupSize[1]); in chooseDispatchSize()
172 workgroupSize = std::min(workgroupSize, maxSquareWorkGroupSize); in chooseDispatchSize()
177 uint32_t power = static_cast<uint32_t>(std::log2(static_cast<float>(workgroupSize))); in chooseDispatchSize()
178 workgroupSize = 1u << power; in chooseDispatchSize()
179 CHECK(workgroupInvocationsX % workgroupSize == 0); in chooseDispatchSize()
180 CHECK(workgroupInvocationsY % workgroupSize == 0); in chooseDispatchSize()
183 .workgroupSize = workgroupSize, in chooseDispatchSize()
[all …]