35 lines
932 B
Python
35 lines
932 B
Python
# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
"""GPU Memory Service server components."""
|
|
|
|
from gpu_memory_service.common.types import (
|
|
GrantedLockType,
|
|
RequestedLockType,
|
|
ServerState,
|
|
StateSnapshot,
|
|
)
|
|
from gpu_memory_service.server.handler import MetadataEntry, RequestHandler
|
|
from gpu_memory_service.server.locking import Connection, GMSLocalFSM
|
|
from gpu_memory_service.server.memory_manager import (
|
|
AllocationInfo,
|
|
AllocationNotFoundError,
|
|
GMSServerMemoryManager,
|
|
)
|
|
from gpu_memory_service.server.rpc import GMSRPCServer
|
|
|
|
__all__ = [
|
|
"GMSRPCServer",
|
|
"GMSServerMemoryManager",
|
|
"AllocationInfo",
|
|
"AllocationNotFoundError",
|
|
"MetadataEntry",
|
|
"Connection",
|
|
"GrantedLockType",
|
|
"RequestedLockType",
|
|
"RequestHandler",
|
|
"ServerState",
|
|
"GMSLocalFSM",
|
|
"StateSnapshot",
|
|
]
|