def _lower_model_to_backend(
        self, mod: torch.fx.GraphModule, inputs: Iterable[torch.Tensor]
    ):
        """
        Lower a GraphModule `mod` to AITemplate with `inputs`.
        """
        # Current code for lowering is place-holder, subject to future change
        # based on feeds model's actual status
        interp = AITInterpreter(mod, [inputs])
        interpreter_result = interp.run(*inputs)
        return AITModule(
            torch.classes.fb.AITModel(
                interpreter_result.engine.lib_path,
                interpreter_result.input_names,
                interpreter_result.output_names,
                torch.float16,
                torch.float,
                1,  # num_runtimes
                False,
            ),
            interpreter_result,
        )

        Lower a GraphModule `mod` to AITemplate with `inputs`.
        
fx2ait/fx2ait/ait_splitter.py