pytorch/torch/utils/serialization/config.py

import sys
from typing import Optional as _Optional, TYPE_CHECKING as _TYPE_CHECKING


if _TYPE_CHECKING:
    from torch.serialization import LoadEndianness as _LoadEndianess

from torch.utils._config_module import install_config_module as _install_config_module


class load:
    mmap: bool = False
    endianness: _Optional["_LoadEndianess"] = None
    # MAP_PRIVATE = 2
    mmap_flags: _Optional[int] = None if sys.platform == "win32" else 2


class save:
    compute_crc32: bool = True
    use_pinned_memory_for_d2h: bool = False


_install_config_module(sys.modules[__name__])