added fix for WorkUCC (#84368)
Added new constructor for WorkUCC to take in optional inputTensors argument for to enable record_shapes=True for profiling purposes. Tested at https://github.com/pytorch/pytorch/pull/84323 which manually merges in https://github.com/pytorch/pytorch/pull/83285.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/84368
Approved by: https://github.com/kingchc, https://github.com/kwen2501
