Fix CPU Unit Test CI on OSS (#1777)

Summary:
Pull Request resolved: #1777

{F1467318350}

CPU Unit Test CI for OSS failing on call to PT2 ExportedProgram: https://github.com/pytorch/torchrec/actions/runs/8233905334/job/22514441384. OSS CI expects .module() instead of directly calling forward

Reviewed By: joshuadeng, gnahzg

Differential Revision: D54751106

fbshipit-source-id: d1c614ec593edaf31ca894c36d2430e06f779f76

torchrec/distributed/tests/test_pt2.py

@@ -132,7 +132,7 @@ def _test_kjt_input_module(
 
             if test_pt2_ir_export:
                 pt2_ir = torch.export.export(EM, inputs, {}, strict=False)
-                pt2_ir_output = pt2_ir(*inputs)
+                pt2_ir_output = pt2_ir.module()(*inputs)
                 assert_close(eager_output, pt2_ir_output)
 
     def test_kjt_split(self) -> None: