fix crashing on cpu (#1791) #2909

Job	Run time
build-and-test-cpp-aarch64 (openblas)	4m 18s
build-and-push-docker-images	31m 4s
build-python-wheels (ubuntu-20.04, auto64)	36m 46s
build-and-test-cpp-x86_64 (ubuntu-20.04, mkl)	8m 35s
build-python-wheels (windows-2019, auto64)	45m 33s
build-and-test-cpp-x86_64 (ubuntu-20.04, dnnl)	7m 15s
check-python-style	10s
build-python-wheels (ubuntu-20.04, aarch64)	1h 20m 55s
build-python-wheels (macos-12, arm64)	7m 31s
test-python-wheels (ubuntu-20.04)	6m 48s
build-and-deploy-docs	27s
test-python-wheels (windows-2019)	3m 59s
publish-python-wheels-on-pypi	0s
	3h 53m 21s

Provide feedback