[pytorch] add manual registration for trace type (#40903)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/40903
This PR continues the work of #38467 - decoupling Autograd and Trace for manually registered ops.
ghstack-source-id: 107158638
Test Plan: CI
Differential Revision: D22354804
fbshipit-source-id: f5ea45ade2850296c62707a2a4449d7d67a9f5b5