[complex] Enable complex autograd and jit tests for `trace` (#51537)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/50381
Now that `index_fill_` supports complex, we can enable complex support for `trace`.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/51537
Reviewed By: H-Huang
Differential Revision: D26198904
Pulled By: anjali411
fbshipit-source-id: d62bb02549919fe35b0bac44f77af964ebd0e92e
