Lines Matching refs:gpu_arch
35 static inline auto gemm_selector_get_data(int gpu_arch, char precision, char trans_A) -> decltype(d… in gemm_selector_get_data() argument
36 if (gpu_arch >= 910) { 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
46 if (gpu_arch >= 800) { 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
104 if (gpu_arch >= 910) { 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
114 if (gpu_arch >= 900) { in nontensor_rtc_get_data()
117 } else if (gpu_arch >= 800) { 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()