icefall/.mypy_cache/3.9/torch/cuda/graphs.data.json

{".class": "MypyFile", "_fullname": "torch.cuda.graphs", "is_partial_stub_package": false, "is_stub": false, "names": {".class": "SymbolTable", "CUDAGraph": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["torch._C._CUDAGraph"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.cuda.graphs.CUDAGraph", "name": "CUDAGraph", "type_vars": []}, "flags": [], "fullname": "torch.cuda.graphs.CUDAGraph", "metaclass_type": null, "metadata": {}, "module_name": "torch.cuda.graphs", "mro": ["torch.cuda.graphs.CUDAGraph", "torch._C._CUDAGraph", "builtins.object"], "names": {".class": "SymbolTable", "__init__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["self"], "flags": [], "fullname": "torch.cuda.graphs.CUDAGraph.__init__", "name": "__init__", "type": null}}, "__new__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["cls"], "flags": [], "fullname": "torch.cuda.graphs.CUDAGraph.__new__", "name": "__new__", "type": null}}, "capture_begin": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1], "arg_names": ["self", "pool"], "flags": [], "fullname": "torch.cuda.graphs.CUDAGraph.capture_begin", "name": "capture_begin", "type": null}}, "capture_end": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["self"], "flags": [], "fullname": "torch.cuda.graphs.CUDAGraph.capture_end", "name": "capture_end", "type": null}}, "pool": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["self"], "flags": [], "fullname": "torch.cuda.graphs.CUDAGraph.pool", "name": "pool", "type": null}}, "replay": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["self"], "flags": [], "fullname": "torch.cuda.graphs.CUDAGraph.replay", "name": "replay", "type": null}}, "reset": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["self"], "flags": [], "fullname": "torch.cuda.graphs.CUDAGraph.reset", "name": "reset", "type": null}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "Graphed@340": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["torch.autograd.function.Function"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.cuda.graphs.Graphed@340", "name": "Graphed", "type_vars": []}, "flags": ["fallback_to_any"], "fullname": "torch.cuda.graphs.Graphed@340", "metaclass_type": null, "metadata": {}, "module_name": "torch.cuda.graphs", "mro": ["torch.cuda.graphs.Graphed@340", "torch.autograd.function.Function", "builtins.object"], "names": {".class": "SymbolTable", "backward": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0, 2], "arg_names": ["ctx", "grads"], "flags": ["is_static", "is_decorated"], "fullname": "torch.cuda.graphs.Graphed@340.backward", "name": "backward", "type": null}, "is_overload": false, "var": {".class": "Var", "flags": ["is_initialized_in_class", "is_staticmethod", "is_ready"], "fullname": null, "name": "backward", "type": {".class": "AnyType", "missing_import_name": null, "source_any": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, "type_of_any": 7}}}}, "forward": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0, 2], "arg_names": ["ctx", "inputs"], "flags": ["is_static", "is_decorated"], "fullname": "torch.cuda.graphs.Graphed@340.forward", "name": "forward", "type": null}, "is_overload": false, "var": {".class": "Var", "flags": ["is_initialized_in_class", "is_staticmethod", "is_ready"], "fullname": null, "name": "forward", "type": {".class": "CallableType", "arg_kinds": [0, 2], "arg_names": ["ctx", "inputs"], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": true, "is_ellipsis_args": false, "name": "forward of Graphed", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, "type_guard": null, "variables": []}}}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "_CUDAGraph": {".class": "SymbolTableNode", "cross_ref": "torch._C._CUDAGraph", "kind": "Gdef"}, "__doc__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.cuda.graphs.__doc__", "name": "__doc__", "type": "builtins.str"}}, "__file__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.cuda.graphs.__file__", "name": "__file__", "type": "builtins.str"}}, "__name__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.cuda.graphs.__name__", "name": "__name__", "type": "builtins.str"}}, "__package__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.cuda.graphs.__package__", "name": "__package__", "type": "builtins.str"}}, "_dummy_type": {".class": "SymbolTableNode", "cross_ref": "torch.cuda._utils._dummy_type", "kind": "Gdef"}, "_graph_pool_handle": {".class": "SymbolTableNode", "cross_ref": "torch._C._graph_pool_handle", "kind": "Gdef"}, "gc": {".class": "SymbolTableNode", "cross_ref": "gc", "kind": "Gdef"}, "graph": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["builtins.object"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.cuda.graphs.graph", "name": "graph", "type_vars": []}, "flags": [], "fullname": "torch.cuda.graphs.graph", "metaclass_type": null, "metadata": {}, "module_name": "torch.cuda.graphs", "mro": ["torch.cuda.graphs.graph", "builtins.object"], "names": {".class": "SymbolTable", "__enter__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["self"], "flags": [], "fullname": "torch.cuda.graphs.graph.__enter__", "name": "__enter__", "type": null}}, "__exit__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 0], "arg_names": ["self", "exc_type", "exc_value", "traceback"], "flags": [], "fullname": "torch.cuda.graphs.graph.__exit__", "name": "__exit__", "type": null}}, "__init__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1], "arg_names": ["self", "cuda_graph", "pool", "stream"], "flags": [], "fullname": "torch.cuda.graphs.graph.__init__", "name": "__init__", "type": null}}, "capture_stream": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.cuda.graphs.graph.capture_stream", "name": "capture_stream", "type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}}}, "cuda_graph": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.cuda.graphs.graph.cuda_graph", "name": "cuda_graph", "type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}}}, "default_capture_stream": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class"], "fullname": "torch.cuda.graphs.graph.default_capture_stream", "name": "default_capture_stream", "type": {".class": "NoneType"}}}, "pool": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.cuda.graphs.graph.pool", "name": "pool", "type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}}}, "stream_ctx": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.cuda.graphs.graph.stream_ctx", "name": "stream_ctx", "type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "graph_pool_handle": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [], "arg_names": [], "flags": [], "fullname": "torch.cuda.graphs.graph_pool_handle", "name": "graph_pool_handle", "type": null}}, "make_graphed_callables": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["callables", "sample_args"], "flags": [], "fullname": "torch.cuda.graphs.make_graphed_callables", "name": "make_graphed_callables", "type": null}}, "torch": {".class": "SymbolTableNode", "cross_ref": "torch", "kind": "Gdef"}}, "path": "/home/patrick/anaconda3/lib/python3.9/site-packages/torch/cuda/graphs.py"}