[Dynamo] Fix dynamo trace rules (#120371)
```test_trace_rules.py``` is still failing due to this.
Fixes https://github.com/pytorch/pytorch/issues/114831
(Having this here will run the disabled test on the PR)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/120371
Approved by: https://github.com/drisspg, https://github.com/huydhn
