It seems all tests in `py/torch_tensorrt/fx/test` are not run in CI at all. It would be good to add them to CI.