MCPcopy Create free account
hub / github.com/NVlabs/tiny-cuda-nn / parallel_for_kernel

Function parallel_for_kernel

include/tiny-cuda-nn/common_host.h:414–419  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

412
413template <typename F>
414__global__ void parallel_for_kernel(const size_t n_elements, F fun) {
415 const size_t i = threadIdx.x + blockIdx.x * blockDim.x;
416 if (i >= n_elements) return;
417
418 fun(i);
419}
420
421template <typename F>
422inline void parallel_for_gpu(uint32_t shmem_size, cudaStream_t stream, size_t n_elements, F&& fun) {

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected