fix crash with return alternative on CUDA (#1733) #2801

Job	Run time
build-and-test-cpp-aarch64 (openblas)	4m 2s
build-and-test-cpp-aarch64 (openblas)	4m 2s
check-python-style	8s
check-python-style	8s
build-and-push-docker-images	1h 48m 59s
build-and-push-docker-images	1h 48m 59s
build-and-test-cpp-x86_64 (ubuntu-20.04, mkl)	5m 36s
build-and-test-cpp-x86_64 (ubuntu-20.04, mkl)	5m 36s
build-python-wheels (ubuntu-20.04, auto64)	2h 5m 29s
build-python-wheels (ubuntu-20.04, auto64)	2h 5m 29s
build-python-wheels (macos-11, auto64)	42m 23s
build-python-wheels (macos-11, auto64)	42m 23s
build-and-test-cpp-x86_64 (ubuntu-20.04, dnnl)	5m 1s
build-and-test-cpp-x86_64 (ubuntu-20.04, dnnl)	5m 1s
build-python-wheels (windows-2019, auto64)	5h 26m 17s
build-python-wheels (windows-2019, auto64)	5h 26m 17s
build-and-test-cpp-x86_64 (macos-11, mkl)	10m 12s
build-and-test-cpp-x86_64 (macos-11, mkl)	10m 12s
build-python-wheels (ubuntu-20.04, aarch64)	1h 14m 57s
build-python-wheels (ubuntu-20.04, aarch64)	1h 14m 57s
build-python-wheels (macos-11, arm64)	12m 39s
build-python-wheels (macos-11, arm64)	12m 39s
build-and-test-cpp-x86_64 (macos-11, dnnl)	10m 9s
build-and-test-cpp-x86_64 (macos-11, dnnl)	10m 9s
test-python-wheels (ubuntu-20.04)	15s
test-python-wheels (ubuntu-20.04)	28s
build-and-deploy-docs	1m 9s
build-and-deploy-docs	1m 9s
test-python-wheels (macos-11)	27s
test-python-wheels (macos-11)	39s
test-python-wheels (windows-2019)	24s
test-python-wheels (windows-2019)	39s
publish-python-wheels-on-pypi	0s
publish-python-wheels-on-pypi	0s
	1d 0h 16m 54s

Provide feedback