Searched refs:base_k (Results 1 – 1 of 1) sorted by relevance
100 #define prefetchIntoRegisters(base_k) \ in EigenContractionKernelInternal() argument121 const Index lhs_horiz_0 = base_k + threadIdx.z + 0 * 8; \ in EigenContractionKernelInternal()122 const Index lhs_horiz_1 = base_k + threadIdx.z + 1 * 8; \ in EigenContractionKernelInternal()123 const Index lhs_horiz_2 = base_k + threadIdx.z + 2 * 8; \ in EigenContractionKernelInternal()124 const Index lhs_horiz_3 = base_k + threadIdx.z + 3 * 8; \ in EigenContractionKernelInternal()125 const Index lhs_horiz_4 = base_k + threadIdx.z + 4 * 8; \ in EigenContractionKernelInternal()126 const Index lhs_horiz_5 = base_k + threadIdx.z + 5 * 8; \ in EigenContractionKernelInternal()127 const Index lhs_horiz_6 = base_k + threadIdx.z + 6 * 8; \ in EigenContractionKernelInternal()128 const Index lhs_horiz_7 = base_k + threadIdx.z + 7 * 8; \ in EigenContractionKernelInternal()177 const Index rhs_vert = base_k + load_idx_vert; \ in EigenContractionKernelInternal()[all …]