25 lines
658 B
Python
25 lines
658 B
Python
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
|
|
calculate_storage_offsets: bool = False
|
|
|
|
|
|
class save:
|
|
compute_crc32: bool = True
|
|
use_pinned_memory_for_d2h: bool = False
|
|
storage_alignment: int = 64
|
|
|
|
|
|
_install_config_module(sys.modules[__name__])
|