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 (115feab3df3b01e57c1003b30e8ed5875e39a742)