Coverage for cuda/pathfinder/_binaries/windows_nsight.py: 100.00%

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

2# SPDX-License-Identifier: Apache-2.0 

3 

4import importlib (empty)

5import os (empty)

6from collections.abc import Iterator (empty)

7from typing import Any (empty)

8 

9from cuda.pathfinder._utils.windows_arch import windows_machine_arch (empty)

10 

11_REGISTRY_ROOT = r"SOFTWARE\NVIDIA Corporation\Installed Products\Nsight" (empty)

12 

13_NSYS_TARGET_DIR_BY_ARCH = { (empty)

14 "x64": "target-windows-x64", 

15 "arm64": "target-windows-armv8", 

16} 

17 

18_NCU_TARGET_DIR_BY_ARCH = { (empty)

19 "x64": os.path.join("target", "windows-desktop-win7-x64"), 

20 "arm64": os.path.join("target", "windows-desktop-win10-t23x-a64"), 

21} 

22 

23 

24def _installed_product_root(product: str) -> str | None: (empty)

25 """Return the active Nsight product installation recorded by its MSI.""" 

26 # ``winreg`` attributes are absent from the type stubs on non-Windows hosts. 

27 winreg: Any = importlib.import_module("winreg") 16 ctx1mnqaijklbcdeofgp

28 

29 access = winreg.KEY_READ | winreg.KEY_WOW64_64KEY 16 ctx1mnqaijklbcdeofgp

30 product_key_path = rf"{_REGISTRY_ROOT}\{product}" 16 ctx1mnqaijklbcdeofgp

31 try: 16 ctx1mnqaijklbcdeofgp

32 product_context = winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, product_key_path, 0, access) 16 ctx1mnqaijklbcdeofgp

33 except FileNotFoundError: 4 ctx1mnqp

34 return None 3 ctx1mnp

35 

36 try: 12 ctx1aijklbcdeofg

37 with product_context as product_key: 12 ctx1aijklbcdeofg

38 current_version, _ = winreg.QueryValueEx(product_key, "CurrentVersion") 12 ctx1aijklbcdeofg

39 if not isinstance(current_version, str) or not current_version.strip(): 11 ctx1aijklbcdefg

40 raise RuntimeError( 4 ctx1ijkl

41 f"Invalid CurrentVersion value {current_version!r} in " 

42 f"Nsight {product!r} registry registration at {product_key_path!r}" 

43 ) 

44 with winreg.OpenKey(product_key, current_version, 0, access) as version_key: 7 ctx1abcdefg

45 install_root, _ = winreg.QueryValueEx(version_key, None) 6 ctx1abcdef

46 except FileNotFoundError as exc: 7 ctx1ijklofg

47 raise RuntimeError(f"Incomplete Nsight {product!r} registry registration at {product_key_path!r}") from exc 3 ctx1ofg

48 

49 if not isinstance(install_root, str) or not install_root.strip(): 5 ctx1abcde

50 raise RuntimeError( 4 ctx1bcde

51 f"Invalid installation directory {install_root!r} in Nsight {product!r} " 

52 f"registry registration at {product_key_path!r} version {current_version!r}" 

53 ) 

54 return install_root 1 ctx1a

55 

56 

57def nsys_candidate_paths() -> Iterator[str]: (empty)

58 install_root = _installed_product_root("Systems") 9 ctx1nzABCGDEF

59 if install_root is None: 9 ctx1nzABCGDEF

60 return 2 ctx1nG

61 

62 target_dir = _NSYS_TARGET_DIR_BY_ARCH[windows_machine_arch()] 7 ctx1zABCDEF

63 yield os.path.join(install_root, target_dir, "nsys.exe") 7 ctx1zABCDEF

64 

65 

66def ncu_candidate_paths() -> Iterator[str]: (empty)

67 install_root = _installed_product_root("Compute") 12 ctx1mHrstuIvwxyJ

68 if install_root is None: 12 ctx1mHrstuIvwxyJ

69 return 2 ctx1mI

70 

71 yield os.path.join(install_root, "ncu.bat") 10 ctx1HrstuvwxyJ

72 

73 target_dir = _NCU_TARGET_DIR_BY_ARCH[windows_machine_arch()] 8 ctx1rstuvwxy

74 yield os.path.join(install_root, target_dir, "ncu.exe") 8 ctx1rstuvwxy