Searched refs:n_batch (Results 1 – 2 of 2) sorted by relevance
| /libCEED/backends/magma/ |
| H A D | ceed-magma-gemm-selector.cpp | 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
59 *n_batch = CEED_AUTOTUNE_GEMM_SELECTOR_N_BATCH; in gemm_selector()
92 *n_batch = (n_ == n_batch_) ? n : n_batch_; in gemm_selector()
95 *n_batch = n; in gemm_selector()
|
| H A D | ceed-magma-gemm-selector.h | 12 …tor(int gpu_arch, char precision, char trans_A, int m, int n, int k, int *n_batch, int *use_magma);
|