def profile(
        self, inputs: List[torch.Tensor], filename: str, num_iters: int
    ) -> None:
        """
        Profile the AIT module and save the report to a file. The AITModule
        must be created with allow_scripting=False.
        inputs: sample inputs
        filename: report filename
        num_iters: number of iterations per op run
        """
        self.engine.profile(inputs, filename, num_iters)

        Profile the AIT module and save the report to a file. The AITModule
        must be created with allow_scripting=False.
        inputs: sample inputs
        filename: report filename
        num_iters: number of iterations per op run
        
fx2ait/fx2ait/ait_module.py