Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/mps/operations/
DMultiTensorApply.h297 uint32_t maxThreadsPerGroup = [fusedOptimizerPSO maxTotalThreadsPerThreadgroup];
298 … MTLSize threadGroupSize = MTLSizeMake(std::min(maxThreadsPerGroup, kThreadGroupSize), 1, 1);
332 uint32_t maxThreadsPerGroup = [fusedOptimizerPSO maxTotalThreadsPerThreadgroup];
333 MTLSize threadGroupSize = MTLSizeMake(std::min(maxThreadsPerGroup, kThreadGroupSize), 1, 1);
DQuantized.mm745 const auto maxThreadsPerGroup = [quantizedPSO maxTotalThreadsPerThreadgroup];
802 …const auto maxThreadsPerGroup = static_cast<decltype(M)>([quantizedPSO maxTotalThreadsPerThreadgro…
/external/pytorch/aten/src/ATen/native/mps/
DOperationUtils.h394 const uint32_t maxThreadsPerGroup = [cplState maxTotalThreadsPerThreadgroup]; in mtl_dispatch1DJob() local
396 auto threadGroupSize = MTLSizeMake(std::min(maxThreadsPerGroup, length), 1, 1); in mtl_dispatch1DJob()