Add CUDAEvent recording for constant folding to show up. (#119216)
Summary: Add a layer of call to let CUDAEvent show up for constant folding.
Test Plan: Existing tests
Differential Revision: D53437934
Pull Request resolved: https://github.com/pytorch/pytorch/pull/119216
Approved by: https://github.com/khabinov