Searched refs:sB (Results 1 – 2 of 2) sorted by relevance
| /libCEED/include/ceed/jit-source/magma/ |
| H A D | magma-basis-interp-deriv-nontensor.h | 26 CeedScalar *sB = shared_data + ty * P * NB; in magma_basis_nontensor_device_n() local
31 read_B_g2s_1D_nosync<CeedScalar, Q, P, NB>(tx, myn, dB, sB); in magma_basis_nontensor_device_n()
41 mul_rAsBrC_1D_nosync<CeedScalar, Q, P, NB>(rA, sB, rC); in magma_basis_nontensor_device_n()
70 CeedScalar *sB = shared_data + ty * Q * NB; in magma_basis_nontensor_device_t() local
83 read_B_g2s_1D_nosync<CeedScalar, P, Q, NB>(tx, myn, dB, sB); in magma_basis_nontensor_device_t()
87 addmul_rAsBrC_1D_nosync<CeedScalar, P, Q, NB>(rA, sB, rC); in magma_basis_nontensor_device_t()
116 CeedScalar *sB = shared_data + ty * Q * NB; in magma_basis_nontensor_device_ta() local
129 read_B_g2s_1D_nosync<CeedScalar, P, Q, NB>(tx, myn, dB, sB); in magma_basis_nontensor_device_ta()
133 addmul_rAsBrC_1D_nosync<CeedScalar, P, Q, NB>(rA, sB, rC); in magma_basis_nontensor_device_ta()
162 CeedScalar *sB = shared_data + ty * P * NB; in magma_basis_nontensor_device_n1() local
[all …]
|
| H A D | magma-common-nontensor.h | 70 …ic __device__ __inline__ void read_B_g2s_1D_nosync(const int tx, const int n, const T *dB, T *sB) { in read_B_g2s_1D_nosync() argument
75 sB[i + tx] = dB[i + tx]; in read_B_g2s_1D_nosync()
80 sB[i + tx] = dB[i + tx]; in read_B_g2s_1D_nosync()
84 sB[i + tx] = dB[i + tx]; in read_B_g2s_1D_nosync()
133 static __device__ __inline__ void mul_rAsBrC_1D_nosync(T rA[Q], T *sB, T rC[NB]) { in mul_rAsBrC_1D_nosync() argument
140 rB[j] = sB[i * Q + j]; in mul_rAsBrC_1D_nosync()
157 static __device__ __inline__ void addmul_rAsBrC_1D_nosync(T rA[Q], T *sB, T rC[NB]) { in addmul_rAsBrC_1D_nosync() argument
164 rB[j] = sB[i * Q + j]; in addmul_rAsBrC_1D_nosync()
|