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

{".class": "MypyFile", "_fullname": "torch.cuda.memory", "is_partial_stub_package": false, "is_stub": false, "names": {".class": "SymbolTable", "Any": {".class": "SymbolTableNode", "cross_ref": "typing.Any", "kind": "Gdef"}, "Device": {".class": "SymbolTableNode", "cross_ref": "torch.types.Device", "kind": "Gdef"}, "Dict": {".class": "SymbolTableNode", "cross_ref": "typing.Dict", "kind": "Gdef"}, "Union": {".class": "SymbolTableNode", "cross_ref": "typing.Union", "kind": "Gdef"}, "__doc__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.cuda.memory.__doc__", "name": "__doc__", "type": "builtins.str"}}, "__file__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.cuda.memory.__file__", "name": "__file__", "type": "builtins.str"}}, "__name__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.cuda.memory.__name__", "name": "__name__", "type": "builtins.str"}}, "__package__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.cuda.memory.__package__", "name": "__package__", "type": "builtins.str"}}, "_free_mutex": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [], "arg_names": [], "flags": ["is_generator", "is_decorated"], "fullname": "torch.cuda.memory._free_mutex", "name": "_free_mutex", "type": null}, "is_overload": false, "var": {".class": "Var", "flags": ["is_ready"], "fullname": null, "name": "_free_mutex", "type": {".class": "CallableType", "arg_kinds": [], "arg_names": [], "arg_types": [], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "_free_mutex", "ret_type": {".class": "Instance", "args": [{".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}], "type_ref": "contextlib._GeneratorContextManager"}, "type_guard": null, "variables": []}}}}, "_get_device_index": {".class": "SymbolTableNode", "cross_ref": "torch.cuda._utils._get_device_index", "kind": "Gdef"}, "_host_allocator": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [], "arg_names": [], "flags": [], "fullname": "torch.cuda.memory._host_allocator", "name": "_host_allocator", "type": null}}, "_lazy_init": {".class": "SymbolTableNode", "cross_ref": "torch.cuda._lazy_init", "kind": "Gdef"}, "caching_allocator_alloc": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1, 1], "arg_names": ["size", "device", "stream"], "flags": [], "fullname": "torch.cuda.memory.caching_allocator_alloc", "name": "caching_allocator_alloc", "type": {".class": "CallableType", "arg_kinds": [0, 1, 1], "arg_names": ["size", "device", "stream"], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "UnionType", "items": ["torch._C.device", "builtins.str", "builtins.int", {".class": "NoneType"}]}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "caching_allocator_alloc", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, "type_guard": null, "variables": []}}}, "caching_allocator_delete": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["mem_ptr"], "flags": [], "fullname": "torch.cuda.memory.caching_allocator_delete", "name": "caching_allocator_delete", "type": null}}, "collections": {".class": "SymbolTableNode", "cross_ref": "collections", "kind": "Gdef"}, "contextlib": {".class": "SymbolTableNode", "cross_ref": "contextlib", "kind": "Gdef"}, "empty_cache": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [], "arg_names": [], "flags": [], "fullname": "torch.cuda.memory.empty_cache", "name": "empty_cache", "type": {".class": "CallableType", "arg_kinds": [], "arg_names": [], "arg_types": [], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "empty_cache", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "is_initialized": {".class": "SymbolTableNode", "cross_ref": "torch.cuda.is_initialized", "kind": "Gdef"}, "list_gpu_processes": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [1], "arg_names": ["device"], "flags": [], "fullname": "torch.cuda.memory.list_gpu_processes", "name": "list_gpu_processes", "type": {".class": "CallableType", "arg_kinds": [1], "arg_names": ["device"], "arg_types": [{".class": "UnionType", "items": ["torch._C.device", "builtins.str", "builtins.int", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "list_gpu_processes", "ret_type": "builtins.str", "type_guard": null, "variables": []}}}, "max_memory_allocated": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [1], "arg_names": ["device"], "flags": [], "fullname": "torch.cuda.memory.max_memory_allocated", "name": "max_memory_allocated", "type": {".class": "CallableType", "arg_kinds": [1], "arg_names": ["device"], "arg_types": [{".class": "UnionType", "items": ["torch._C.device", "builtins.str", "builtins.int", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "max_memory_allocated", "ret_type": "builtins.int", "type_guard": null, "variables": []}}}, "max_memory_cached": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [1], "arg_names": ["device"], "flags": [], "fullname": "torch.cuda.memory.max_memory_cached", "name": "max_memory_cached", "type": {".class": "CallableType", "arg_kinds": [1], "arg_names": ["device"], "arg_types": [{".class": "UnionType", "items": ["torch._C.device", "builtins.str", "builtins.int", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "max_memory_cached", "ret_type": "builtins.int", "type_guard": null, "variables": []}}}, "max_memory_reserved": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [1], "arg_names": ["device"], "flags": [], "fullname": "torch.cuda.memory.max_memory_reserved", "name": "max_memory_reserved", "type": {".class": "CallableType", "arg_kinds": [1], "arg_names": ["device"], "arg_types": [{".class": "UnionType", "items": ["torch._C.device", "builtins.str", "builtins.int", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "max_memory_reserved", "ret_type": "builtins.int", "type_guard": null, "variables": []}}}, "mem_get_info": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [1], "arg_names": ["device"], "flags": [], "fullname": "torch.cuda.memory.mem_get_info", "name": "mem_get_info", "type": {".class": "CallableType", "arg_kinds": [1], "arg_names": ["device"], "arg_types": [{".class": "UnionType", "items": ["torch._C.device", "builtins.str", "builtins.int", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "mem_get_info", "ret_type": "builtins.int", "type_guard": null, "variables": []}}}, "memory_allocated": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [1], "arg_names": ["device"], "flags": [], "fullname": "torch.cuda.memory.memory_allocated", "name": "memory_allocated", "type": {".class": "CallableType", "arg_kinds": [1], "arg_names": ["device"], "arg_types": [{".class": "UnionType", "items": ["torch._C.device", "builtins.str", "builtins.int", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "memory_allocated", "ret_type": "builtins.int", "type_guard": null, "variables": []}}}, "memory_cached": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [1], "arg_names": ["device"], "flags": [], "fullname": "torch.cuda.memory.memory_cached", "name": "memory_cached", "type": {".class": "CallableType", "arg_kinds": [1], "arg_names": ["device"], "arg_types": [{".class": "UnionType", "items": ["torch._C.device", "builtins.str", "builtins.int", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "memory_cached", "ret_type": "builtins.int", "type_guard": null, "variables": []}}}, "memory_reserved": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [1], "arg_names": ["device"], "flags": [], "fullname": "torch.cuda.memory.memory_reserved", "name": "memory_reserved", "type": {".class": "CallableType", "arg_kinds": [1], "arg_names": ["device"], "arg_types": [{".class": "UnionType", "items": ["torch._C.device", "builtins.str", "builtins.int", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "memory_reserved", "ret_type": "builtins.int", "type_guard": null, "variables": []}}}, "memory_snapshot": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [], "arg_names": [], "flags": [], "fullname": "torch.cuda.memory.memory_snapshot", "name": "memory_snapshot", "type": null}}, "memory_stats": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [1], "arg_names": ["device"], "flags": [], "fullname": "torch.cuda.memory.memory_stats", "name": "memory_stats", "type": {".class": "CallableType", "arg_kinds": [1], "arg_names": ["device"], "arg_types": [{".class": "UnionType", "items": ["torch._C.device", "builtins.str", "builtins.int", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "memory_stats", "ret_type": {".class": "Instance", "args": ["builtins.str", {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "type_ref": "builtins.dict"}, "type_guard": null, "variables": []}}}, "memory_stats_as_nested_dict": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [1], "arg_names": ["device"], "flags": [], "fullname": "torch.cuda.memory.memory_stats_as_nested_dict", "name": "memory_stats_as_nested_dict", "type": {".class": "CallableType", "arg_kinds": [1], "arg_names": ["device"], "arg_types": [{".class": "UnionType", "items": ["torch._C.device", "builtins.str", "builtins.int", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "memory_stats_as_nested_dict", "ret_type": {".class": "Instance", "args": ["builtins.str", {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "type_ref": "builtins.dict"}, "type_guard": null, "variables": []}}}, "memory_summary": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [1, 1], "arg_names": ["device", "abbreviated"], "flags": [], "fullname": "torch.cuda.memory.memory_summary", "name": "memory_summary", "type": {".class": "CallableType", "arg_kinds": [1, 1], "arg_names": ["device", "abbreviated"], "arg_types": [{".class": "UnionType", "items": ["torch._C.device", "builtins.str", "builtins.int", {".class": "NoneType"}]}, "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "memory_summary", "ret_type": "builtins.str", "type_guard": null, "variables": []}}}, "reset_accumulated_memory_stats": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [1], "arg_names": ["device"], "flags": [], "fullname": "torch.cuda.memory.reset_accumulated_memory_stats", "name": "reset_accumulated_memory_stats", "type": {".class": "CallableType", "arg_kinds": [1], "arg_names": ["device"], "arg_types": [{".class": "UnionType", "items": ["torch._C.device", "builtins.str", "builtins.int", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "reset_accumulated_memory_stats", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "reset_max_memory_allocated": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [1], "arg_names": ["device"], "flags": [], "fullname": "torch.cuda.memory.reset_max_memory_allocated", "name": "reset_max_memory_allocated", "type": {".class": "CallableType", "arg_kinds": [1], "arg_names": ["device"], "arg_types": [{".class": "UnionType", "items": ["torch._C.device", "builtins.str", "builtins.int", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "reset_max_memory_allocated", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "reset_max_memory_cached": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [1], "arg_names": ["device"], "flags": [], "fullname": "torch.cuda.memory.reset_max_memory_cached", "name": "reset_max_memory_cached", "type": {".class": "CallableType", "arg_kinds": [1], "arg_names": ["device"], "arg_types": [{".class": "UnionType", "items": ["torch._C.device", "builtins.str", "builtins.int", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "reset_max_memory_cached", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "reset_peak_memory_stats": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [1], "arg_names": ["device"], "flags": [], "fullname": "torch.cuda.memory.reset_peak_memory_stats", "name": "reset_peak_memory_stats", "type": {".class": "CallableType", "arg_kinds": [1], "arg_names": ["device"], "arg_types": [{".class": "UnionType", "items": ["torch._C.device", "builtins.str", "builtins.int", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "reset_peak_memory_stats", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "set_per_process_memory_fraction": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1], "arg_names": ["fraction", "device"], "flags": [], "fullname": "torch.cuda.memory.set_per_process_memory_fraction", "name": "set_per_process_memory_fraction", "type": {".class": "CallableType", "arg_kinds": [0, 1], "arg_names": ["fraction", "device"], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "UnionType", "items": ["torch._C.device", "builtins.str", "builtins.int", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "set_per_process_memory_fraction", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "torch": {".class": "SymbolTableNode", "cross_ref": "torch", "kind": "Gdef"}, "warnings": {".class": "SymbolTableNode", "cross_ref": "warnings", "kind": "Gdef"}}, "path": "/home/patrick/anaconda3/lib/python3.9/site-packages/torch/cuda/memory.py"}