Flash attention #2724

Job	Run time
build-and-push-docker-images	1h 52m 9s
build-and-test-cpp-x86_64 (ubuntu-20.04, mkl)	6m 26s
build-python-wheels (ubuntu-20.04, auto64)	2h 0m 57s
build-and-test-cpp-x86_64 (ubuntu-20.04, dnnl)	5m 6s
build-and-test-cpp-aarch64 (openblas)	4m 30s
build-python-wheels (macos-11, auto64)	39m 38s
build-and-test-cpp-x86_64 (macos-11, mkl)	11m 43s
build-python-wheels (windows-2019, auto64)	5h 27m 44s
check-python-style	10s
build-and-test-cpp-x86_64 (macos-11, dnnl)	10m 55s
build-python-wheels (ubuntu-20.04, aarch64)	1h 14m 10s
build-python-wheels (macos-11, arm64)	11m 55s
test-python-wheels (ubuntu-20.04)	9m 23s
build-and-deploy-docs	1m 5s
test-python-wheels (macos-11)	4m 7s
test-python-wheels (windows-2019)	4m 46s
publish-python-wheels-on-pypi	0s
	12h 24m 44s

Provide feedback