Home
last modified time | relevance | path

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

/libCEED/include/ceed/jit-source/magma/
H A Dmagma-basis-interp-deriv-nontensor.h37 CeedScalar rA[P]; in magma_basis_nontensor_device_n() local
38 …ans_g2r_1D_nosync<CeedScalar, Q, P, MAGMA_BASIS_NTCOL(Q, MAGMA_MAXTHREADS_1D)>(tx, ty, dA, sA, rA); in magma_basis_nontensor_device_n()
41 mul_rAsBrC_1D_nosync<CeedScalar, Q, P, NB>(rA, sB, rC); in magma_basis_nontensor_device_n()
77 CeedScalar rA[Q]; in magma_basis_nontensor_device_t() local
78 …ans_g2r_1D_nosync<CeedScalar, P, Q, MAGMA_BASIS_NTCOL(P, MAGMA_MAXTHREADS_1D)>(tx, ty, dA, sA, rA); in magma_basis_nontensor_device_t()
87 addmul_rAsBrC_1D_nosync<CeedScalar, P, Q, NB>(rA, sB, rC); in magma_basis_nontensor_device_t()
123 CeedScalar rA[Q]; in magma_basis_nontensor_device_ta() local
124 …ans_g2r_1D_nosync<CeedScalar, P, Q, MAGMA_BASIS_NTCOL(P, MAGMA_MAXTHREADS_1D)>(tx, ty, dA, sA, rA); in magma_basis_nontensor_device_ta()
133 addmul_rAsBrC_1D_nosync<CeedScalar, P, Q, NB>(rA, sB, rC); in magma_basis_nontensor_device_ta()
165 CeedScalar rA[P]; in magma_basis_nontensor_device_n1() local
[all …]
H A Dmagma-common-nontensor.h20 …line__ void read_A_notrans_g2r_1D_nosync(const int tx, const int ty, const T *dA, T *sA, T rA[Q]) { in read_A_notrans_g2r_1D_nosync()
35 rA[j] = sA[j * P + tx]; in read_A_notrans_g2r_1D_nosync()
45 …inline__ void read_A_trans_g2r_1D_nosync(const int tx, const int ty, const T *dA, T *sA, T rA[Q]) { in read_A_trans_g2r_1D_nosync()
60 rA[j] = sA[tx * Q + j]; in read_A_trans_g2r_1D_nosync()
133 static __device__ __inline__ void mul_rAsBrC_1D_nosync(T rA[Q], T *sB, T rC[NB]) { in mul_rAsBrC_1D_nosync()
145 rC[i] += rA[j] * rB[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()
168 rC[i] += rA[j] * rB[j]; in addmul_rAsBrC_1D_nosync()