metal : check for BF16 support in concat kernel (#24747)

This commit is contained in:
Georgi Gerganov 2026-06-18 09:16:06 +03:00 committed by GitHub
parent f3e1828164
commit cae0a3b0b0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -7557,7 +7557,9 @@ typedef decltype(kernel_concat<float>) kernel_concat_t;
template [[host_name("kernel_concat_f32")]] kernel kernel_concat_t kernel_concat<float>;
template [[host_name("kernel_concat_f16")]] kernel kernel_concat_t kernel_concat<half>;
#if defined(GGML_METAL_HAS_BF16)
template [[host_name("kernel_concat_bf16")]] kernel kernel_concat_t kernel_concat<bfloat>;
#endif
template [[host_name("kernel_concat_i8")]] kernel kernel_concat_t kernel_concat<char>;
template [[host_name("kernel_concat_i16")]] kernel kernel_concat_t kernel_concat<short>;
template [[host_name("kernel_concat_i32")]] kernel kernel_concat_t kernel_concat<int>;