[torch][segment_reduce] Add missing cuda kernel launch check (#60114)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/60114
Same as title.
Test Plan: Unit test (test_kernel_launch_checks.py) is passing.
Reviewed By: ngimel
Differential Revision: D29169538
fbshipit-source-id: ba4518dcb1a4713144d92faec2bb5bdf656ff7c5
