| 18 | |
| 19 | template <typename... Args> |
| 20 | auto get_quantized_kernel_wrapped( |
| 21 | metal::Device& d, |
| 22 | const std::string& name, |
| 23 | const std::string& func, |
| 24 | const std::string& mode, |
| 25 | const std::string& type, |
| 26 | int group_size, |
| 27 | int bits, |
| 28 | Args... args) { |
| 29 | std::string template_def; |
| 30 | std::string fname = ((mode == "affine") ? "affine_" : "fp_") + func; |
| 31 | template_def = get_template_definition( |
| 32 | name, fname, type, group_size, bits, std::forward<Args>(args)...); |
| 33 | return get_quantized_kernel(d, name, template_def, mode); |
| 34 | } |
| 35 | |
| 36 | template <typename... Args> |
| 37 | auto get_qmm_nax_kernel_wrapped( |
no test coverage detected