[BugFix] Fix update in serial / parallel env #9115

Job	Run time
build-wheel-linux (3.8, cp38-cp38, --index-url https://download.pytorch.org/whl/cpu, "['cpu', '11...	1m 32s
build-wheel-linux (3.9, cp39-cp39, --index-url https://download.pytorch.org/whl/cpu, "['cpu', '11...	1m 34s
build-wheel-mac (3.8, 3.8)	3m 27s
build-wheel-linux (3.10, cp310-cp310, --index-url https://download.pytorch.org/whl/cpu, "['cpu', ...	1m 46s
build-wheel-windows (3.8, 3.8)	3m 41s
build-wheel-mac (3.9, 3.9)	1m 52s
build-wheel-linux (3.11, cp311-cp311, --index-url https://download.pytorch.org/whl/cpu, "['cpu', ...	1m 34s
build-wheel-windows (3.9, 3.9)	2m 57s
build-wheel-windows (3.10, 3.10.3)	3m 51s
build-wheel-windows (3.11, 3.11)	3m 53s
build-wheel-mac (3.10, 3.10.3)	1m 59s
build-wheel-mac (3.11, 3.11)	2m 16s
test-wheel (linux, ubuntu-20.04, 3.8)	2m 23s
test-wheel (linux, ubuntu-20.04, 3.9)	2m 8s
test-wheel (linux, ubuntu-20.04, 3.10)	2m 11s
test-wheel (linux, ubuntu-20.04, 3.11)	2m 11s
test-wheel (mac, macos-11, 3.8)	5m 38s
test-wheel (mac, macos-11, 3.9)	2m 50s
test-wheel (mac, macos-11, 3.10)	2m 54s
test-wheel (mac, macos-11, 3.11)	4m 19s
test-wheel-windows (3.8)	5m 31s
test-wheel-windows (3.9)	5m 24s
test-wheel-windows (3.10)	5m 33s
test-wheel-windows (3.11)	5m 47s
	1h 17m 11s

Provide feedback