Home
last modified time | relevance | path

Searched refs:trans_A (Results 1 – 4 of 4) sorted by relevance

/libCEED/backends/magma/ !
H A Dceed-magma-gemm-selector.cpp35 static inline auto gemm_selector_get_data(int gpu_arch, char precision, char trans_A) -> decltype(d… in gemm_selector_get_data() argument
38 …return (precision == 's') ? ((trans_A == 'n') ? sgemm_nn_mi250x : sgemm_tn_mi250x) : ((trans_A == … in gemm_selector_get_data()
41 …return (precision == 's') ? ((trans_A == 'n') ? sgemm_nn_mi100 : sgemm_tn_mi100) : ((trans_A == 'n… in gemm_selector_get_data()
45 static inline auto gemm_selector_get_data(int gpu_arch, char precision, char trans_A) -> decltype(d… in gemm_selector_get_data() argument
48 …return (precision == 's') ? ((trans_A == 'n') ? sgemm_nn_a100 : sgemm_tn_a100) : ((trans_A == 'n')… in gemm_selector_get_data()
51 …return (precision == 's') ? ((trans_A == 'n') ? sgemm_nn_v100 : sgemm_tn_v100) : ((trans_A == 'n')… in gemm_selector_get_data()
57 void gemm_selector(int gpu_arch, char precision, char trans_A, int m, int n, int k, int *n_batch, i… in gemm_selector() argument
62 const auto &data = gemm_selector_get_data(gpu_arch, precision, trans_A); in gemm_selector()
103 static inline auto nontensor_rtc_get_data(int gpu_arch, char trans_A) -> decltype(drtc_n_mi100) { in nontensor_rtc_get_data() argument
106 return (trans_A == 'n') ? drtc_n_mi250x : drtc_t_mi250x; in nontensor_rtc_get_data()
[all …]
H A Dceed-magma-gemm-nontensor.cpp24 static inline int magmablas_gemm(magma_trans_t trans_A, magma_trans_t trans_B, magma_int_t m, magma… in magmablas_gemm() argument
28 …magmablas_sgemm(trans_A, trans_B, m, n, k, (float)alpha, (const float *)d_A, ldda, (const float *)… in magmablas_gemm()
31 …magmablas_dgemm(trans_A, trans_B, m, n, k, (double)alpha, (const double *)d_A, ldda, (const double… in magmablas_gemm()
38 static inline int magmablas_gemm_batched_strided(magma_trans_t trans_A, magma_trans_t trans_B, magm… in magmablas_gemm_batched_strided() argument
43 …magmablas_sgemm_batched_strided(trans_A, trans_B, m, n, k, (float)alpha, (const float *)d_A, ldda,… in magmablas_gemm_batched_strided()
46 …magmablas_dgemm_batched_strided(trans_A, trans_B, m, n, k, (double)alpha, (const double *)d_A, ldd… in magmablas_gemm_batched_strided()
53 static inline int devblas_gemm(magma_trans_t trans_A, magma_trans_t trans_B, magma_int_t m, magma_i… in devblas_gemm() argument
57 …magma_sgemm(trans_A, trans_B, m, n, k, (float)alpha, (const float *)d_A, ldda, (const float *)d_B,… in devblas_gemm()
59 …magma_dgemm(trans_A, trans_B, m, n, k, (double)alpha, (const double *)d_A, ldda, (const double *)d… in devblas_gemm()
66 static inline int devblas_gemm_batched_strided(magma_trans_t trans_A, magma_trans_t trans_B, magma_… in devblas_gemm_batched_strided() argument
[all …]
H A Dceed-magma-gemm-selector.h12 CEED_INTERN void gemm_selector(int gpu_arch, char precision, char trans_A, int m, int n, int k, int…
15 CEED_INTERN CeedInt nontensor_rtc_get_nb(int gpu_arch, char trans_A, int q_comp, int P, int Q, int …
H A Dceed-magma-gemm-nontensor.h12 CEED_INTERN int magma_gemm_nontensor(magma_trans_t trans_A, magma_trans_t trans_B, magma_int_t m, m…