pt: fix multitask stuck on multiple-gpu #6131

Job	Run time
Build source distribution	15s
determine-arm64-runner	1s
Build wheels for cp311-manylinux_x86_64	5m 38s
Build wheels for cp311-manylinux_x86_64	5m 23s
Build wheels for cp311-macosx_x86_64	10m 36s
Build wheels for cp311-macosx_arm64	3m 25s
Build wheels for cp311-win_amd64	5m 25s
Build wheels for cp310-manylinux_aarch64	5m 20s
build_pypi_index	10s
build_docker (12)	3m 49s
build_docker (_cu11, 11)	4m 30s
upload_pypi	0s
deploy_pypi_index	0s
Pass testing build wheels	3s
	44m 35s

Provide feedback