metal : optimize FA kernels (#10171)

Browse source

* ggml : add ggml_flash_attn_ext_get_prec

* metal : use F16 precision in FA kernels

ggml-ci

* metal : minor clean-up

* metal : compile-guard bf16 FA kernels

ggml-ci

* build : remove obsolete compile flag [no ci]

* metal : prevent int overflows [no ci]

* cuda : disable BF16 FA

ggml-ci

* metal : fix BF16 requirement for FA kernels

ggml-ci

* make : clean-up [no ci]

...

This commit is contained in:

Georgi Gerganov 2024-11-08 13:47:22 +02:00 • committed by GitHub

parent d05b3127bd

commit 841f27abdb

No known key found for this signature in database

GPG key ID: B5690EEEBB952194

8 changed files with 498 additions and 339 deletions

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Download patch file Download diff file Expand all files Collapse all files

733

ggml/src/ggml-metal.metal

View file

File diff suppressed because it is too large Load diff