Compare commits

...

2 Commits

Author SHA1 Message Date
Niko00
4595fb88bd
Merge 6ff2489510 into b08debceca 2026-07-06 12:39:36 +08:00
Niko00
6ff2489510
Add check for WSL_DISTRO_NAME for is_wsl()
WSL supports custom kernels. Add check for WSL_DISTRO_NAME to check custom kernels.
2026-03-01 13:56:19 +08:00

View File

@ -188,6 +188,8 @@ def is_wsl():
return True
elif version.endswith("microsoft-standard-WSL2"):
return True
if os.environ.get("WSL_DISTRO_NAME"):
return True
return False
def get_torch_device():