Searched refs:gpu_arch (Results 1 – 2 of 2) 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() 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()
[all …]
|
| H A D | ceed-magma-gemm-selector.h | 12 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 …
|