There was an error while loading. Please reload this page.
1 parent b5ddb10 commit 425ff30Copy full SHA for 425ff30
1 file changed
nsight/collection/ncu.py
@@ -142,7 +142,14 @@ def try_init_injection() -> None:
142
check_ncu_version(ncu_path)
143
144
inj_dir = get_injection_library_path(ncu_path)
145
- inj_lib_path = os.path.join(inj_dir, "libcuda-injection.so")
+
146
+ if os.name == "posix":
147
+ inj_lib_path = os.path.join(inj_dir, "libcuda-injection.so")
148
+ elif os.name == "nt":
149
+ inj_lib_path = os.path.join(inj_dir, "cuda-injection.dll")
150
+ else:
151
+ raise exceptions.ProfilerException(f"Unsupported operating system: {os.name}")
152
153
if not os.path.isfile(inj_lib_path):
154
raise exceptions.ProfilerException("Failed to find NCU injection library")
155
0 commit comments