Lines Matching refs:trans_A
35 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()
109 return (trans_A == 'n') ? drtc_n_mi100 : drtc_t_mi100; in nontensor_rtc_get_data()
113 static inline auto nontensor_rtc_get_data(int gpu_arch, char trans_A) -> decltype(drtc_n_v100) { in nontensor_rtc_get_data() argument
116 return (trans_A == 'n') ? drtc_n_h100 : drtc_t_h100; in nontensor_rtc_get_data()
119 return (trans_A == 'n') ? drtc_n_a100 : drtc_t_a100; in nontensor_rtc_get_data()
122 return (trans_A == 'n') ? drtc_n_v100 : drtc_t_v100; in nontensor_rtc_get_data()
128 CeedInt nontensor_rtc_get_nb(int gpu_arch, char trans_A, int q_comp, int P, int Q, int N) { in nontensor_rtc_get_nb() argument
132 const auto &data = nontensor_rtc_get_data(gpu_arch, trans_A); in nontensor_rtc_get_nb()