Searched defs:gpu_arch (Results 1 – 1 of 1) sorted by relevance
| /libCEED/backends/magma/ |
| H A D | ceed-magma-gemm-selector.cpp | 35 static inline auto gemm_selector_get_data(int gpu_arch, char precision, char trans_A) -> decltype(d… 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()
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()
103 static inline auto nontensor_rtc_get_data(int gpu_arch, char trans_A) -> decltype(drtc_n_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()
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()
|