Ensure synchronization by adding cuda.synchronize()

[KumoLiu]

Fixes #8054

Description

Add cuda sync after invoke cuda

Types of changes

• Non-breaking change (fix or new feature that would not break existing functionality).
• Breaking change (fix or new feature that would cause existing functionality to change).
• New tests added to cover the changes.
• Integration tests passed locally by running ./runtests.sh -f -u --net --coverage.
• Quick tests passed locally by running ./runtests.sh --quick --unittests --disttests.
• In-line docstrings updated.
• Documentation updated, tested make html command in the docs/ folder.

[ericspod]

Will this be fine if we don't have CUDA installed, ie. CPU-only Pytorch? If not we may have to wrap the call in a function which checks if CUDA is present. It looks fine to me otherwise.

[KumoLiu]

Will this be fine if we don't have CUDA installed, ie. CPU-only Pytorch? If not we may have to wrap the call in a function which checks if CUDA is present. It looks fine to me otherwise.

Hi @ericspod, I add cuda check for the filter one, for the distance culculate one should be fine since it will check cupy before move data to the cuda.

[ericspod]

Hi @ericspod, I add cuda check for the filter one, for the distance culculate one should be fine since it will check cupy before move data to the cuda

Thanks, looks good!

[KumoLiu]

/build