[jit] Add tests for JIT with dynamic shape fusion (#72201)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/72201
Reviewed By: mikaylagawarecki
Differential Revision: D34067211
Pulled By: navahgar
fbshipit-source-id: 2c13bb43c76c7fed720ad37892d2177c3dc0b924
(cherry picked from commit eed2d8cea4a4037bf7784e6ca09e69f63faad6d6)