Method get_context

torch/_dynamo/code_context.py:13–18

(self, code: types.CodeType)

        return code in self.code_context

    def get_context(self, code: types.CodeType):
        ctx = self.code_context.get(code)
        if ctx is None:
            ctx = {}
            self.code_context[code] = ctx
        return ctx

    def pop_context(self, code: types.CodeType):
        ctx = self.get_context(code)