Searched refs:gridDim (Results 1 – 8 of 8) sorted by relevance
28 out[i++] = gridDim.x; in kernel()29 gridDim.x = 0; // expected-error {{no setter defined for property 'x'}} in kernel()30 out[i++] = gridDim.y; in kernel()31 gridDim.y = 0; // expected-error {{no setter defined for property 'y'}} in kernel()32 out[i++] = gridDim.z; in kernel()33 gridDim.z = 0; // expected-error {{no setter defined for property 'z'}} in kernel()
21 out[i++] = gridDim.x; // CHECK: call i32 @llvm.nvvm.read.ptx.sreg.nctaid.x() in kernel()22 out[i++] = gridDim.y; // CHECK: call i32 @llvm.nvvm.read.ptx.sreg.nctaid.y() in kernel()23 out[i++] = gridDim.z; // CHECK: call i32 @llvm.nvvm.read.ptx.sreg.nctaid.z() in kernel()
116 const Index num_threads = blockDim.x * gridDim.x; in ReductionInitKernel()130 if (gridDim.x == 1) { in FullReductionKernel()158 eigen_assert(gridDim.x == 1 || *semaphore >= 2u); in FullReductionKernel()178 if (gridDim.x > 1 && threadIdx.x == 0) { in FullReductionKernel()180 atomicInc(semaphore, gridDim.x + 1); in FullReductionKernel()193 eigen_assert(gridDim.x == 1); in ReductionInitFullReduxKernelHalfFloat()206 const Index num_threads = blockDim.x * gridDim.x; in ReductionInitKernelHalfFloat()226 if (gridDim.x == 1 && first_index == 0) { in FullReductionKernelHalfFloat()256 if (gridDim.x == 1 && first_index == 0) { in FullReductionKernelHalfFloat()380 eigen_assert(gridDim.y == 1);[all …]
25 gridDim.x * blockDim.x * (blockIdx.y * blockDim.y + threadIdx.y); in get_random_seed()
240 const Index step_size = blockDim.x * gridDim.x;
585 const int plane_stride = blockDim.y * gridDim.y;639 const int plane_stride = blockDim.z * gridDim.z;
115 __CUDA_BUILTIN_VAR __cuda_builtin_gridDim_t gridDim; variable
231 ``gridDim`` ``@llvm.nvvm.read.ptx.sreg.nctaid.*``568 ``i32 @llvm.nvvm.read.ptx.sreg.nctaid.{x,y,z}`` gridDim.{x,y,z}