Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DNanCheck.cu30 const size_t maxNumThreadsPerBlock = 256; in checkForNan() local
33 std::min<size_t>(maxNumThreadsPerBlock, tensor.numel()); in checkForNan()