swift-coreml-diffusers

Commit Graph

Author	SHA1	Message	Date
Pedro Cuenca	ed15f78628	Decide attention and ANE on hardware capabilities (#29 ) For Macs with >= 8 performance cores, we select CPU+GPU (original attention). Otherwise we select CPU+ANE (split einsum). Some computers (M1 Pro, 16 core GPU) might yield slightly better performance using CPU+GPU+ANE with SPLIT_EINSUM.	2 years ago

Author

SHA1

Message

Date

Pedro Cuenca

ed15f78628

Decide attention and ANE on hardware capabilities (#29 )

For Macs with >= 8 performance cores, we select CPU+GPU (original
attention). Otherwise we select CPU+ANE (split einsum).

Some computers (M1 Pro, 16 core GPU) might yield slightly better
performance using CPU+GPU+ANE with SPLIT_EINSUM.

1 Commits (04b918ae49c9f6c760aaa6851528df670a20a6f3)