Fix Compilation with Flash Attention #2969

Job	Run time
build-and-push-docker-images	32m 29s
build-and-test-cpp-aarch64 (openblas)	4m 44s
build-python-wheels (ubuntu-20.04, auto64)	36m 32s
build-python-wheels (windows-2019, auto64)	46m 46s
build-and-test-cpp-x86_64 (ubuntu-20.04, mkl)	7m 46s
build-python-wheels (ubuntu-20.04, aarch64)	1h 34m 18s
build-and-test-cpp-x86_64 (ubuntu-20.04, dnnl)	6m 5s
check-python-style	17s
build-python-wheels (macos-13, arm64)	11m 46s
build-and-deploy-docs	34s
test-python-wheels (ubuntu-20.04)	7m 48s
test-python-wheels (windows-2019)	5m 13s
publish-python-wheels-on-pypi	0s
	4h 14m 18s

Provide feedback