chore: optimize compute_smooth_weight #10227

Job	Run time
Build source distribution	13s
determine-arm64-runner	1s
Build wheels for cp311-manylinux_x86_64	6m 14s
Build wheels for cp311-manylinux_x86_64	6m 19s
Build wheels for cp311-macosx_x86_64	4m 43s
Build wheels for cp311-macosx_arm64	3m 13s
Build wheels for cp311-win_amd64	5m 29s
Build wheels for cp310-manylinux_aarch64	6m 9s
build_pypi_index	8s
build_docker (12)	2m 4s
build_docker (_cu11, 11)	3m 9s
upload_pypi	0s
deploy_pypi_index	0s
Pass testing build wheels	4s
	37m 46s

Provide feedback