Coverage for cuda / pathfinder / _dynamic_libs / platform_loader.py: 100.00%

1# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. 

2# SPDX-License-Identifier: Apache-2.0 

3 

4"""Platform loader seam for OS-specific dynamic linking. 

5 

6This module provides a small interface that hides the Linux vs Windows 

7implementation details of: 

8 

9- already-loaded checks 

10- system-search loading 

11- absolute-path loading 

12 

13The orchestration logic in :mod:`load_nvidia_dynamic_lib` should not need to 

14branch on platform; it calls through the loader instance exported here. 

15""" 

16 

17from __future__ import annotations (empty)

18 

19from typing import Protocol (empty)

20 

21from cuda.pathfinder._dynamic_libs.lib_descriptor import LibDescriptor (empty)

22from cuda.pathfinder._dynamic_libs.load_dl_common import LoadedDL (empty)

23from cuda.pathfinder._utils.platform_aware import IS_WINDOWS (empty)

24 

25 

26class PlatformLoader(Protocol): (empty)

27 def check_if_already_loaded_from_elsewhere(self, desc: LibDescriptor, have_abs_path: bool) -> LoadedDL | None: ... (empty)

28 

29 def load_with_system_search(self, desc: LibDescriptor) -> LoadedDL | None: ... (empty)

30 

31 def load_with_abs_path(self, desc: LibDescriptor, found_path: str, found_via: str | None = None) -> LoadedDL: ... (empty)

32 

33 

34if IS_WINDOWS: (empty)

35 from cuda.pathfinder._dynamic_libs import load_dl_windows as _impl (empty)

36else: 

37 from cuda.pathfinder._dynamic_libs import load_dl_linux as _impl (empty)

38 

39# The platform modules already expose functions matching the PlatformLoader 

40# protocol. Wrap in a simple namespace so callers use LOADER.method() syntax. 

41LOADER: PlatformLoader = _impl (empty)