Searched defs:NB (Results 1 – 2 of 2) sorted by relevance
| /libCEED/include/ceed/jit-source/magma/ |
| H A D | magma-common-nontensor.h | 94 static __device__ __inline__ void write_C_r2g_1D_nosync(const int tx, const int n, T rC[NB], T *dC)… in write_C_r2g_1D_nosync()
113 static __device__ __inline__ void sum_C_r2g_1D_nosync(const int tx, const int n, T rC[NB], T *dC) { in sum_C_r2g_1D_nosync()
133 static __device__ __inline__ void mul_rAsBrC_1D_nosync(T rA[Q], T *sB, T rC[NB]) { 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()
|
| /libCEED/backends/magma/ |
| H A D | ceed-magma-basis.c | 382 CeedInt q_comp, NB, M, K; in CeedBasisApplyNonTensorCore_Magma() local
|