basic windows native pytorch support

This commit is contained in:
Seunghoon Lee
2025-05-09 22:23:07 +09:00
parent 51716bbaba
commit 45c0bd6ec6
4 changed files with 136 additions and 123 deletions
+7 -3
View File
@@ -1281,9 +1281,13 @@ mem_mon = modules.memmon.MemUsageMonitor("MemMon", devices.device)
history = history.History()
if devices.backend == "directml":
directml_do_hijack()
elif devices.backend == "zluda":
from modules.zluda import initialize_zluda
initialize_zluda()
elif sys.platform == "win32" and (devices.backend == "zluda" or devices.backend == "rocm"):
from modules.rocm_triton_windows import apply_triton_patches
apply_triton_patches()
if devices.backend == "zluda":
from modules.zluda import initialize_zluda
initialize_zluda()
try:
log.info(f'Device: {print_dict(devices.get_gpu_info())}')
except Exception as ex: