Class WrapperModule

fx/native_interpreter/use_interpreter.py:82–85

    # ===== Stage 3: Create a wrapper GraphModule around the interpreter =====
    class WrapperModule(torch.nn.Module):
        def __init__(self, interpreter):
            super().__init__()
            self.interpreter = interpreter

    wrapper = WrapperModule(interpreter)