zluda init

Author: lshqqytiger

modules/launch_utils.py

@@ -451,7 +451,7 @@ def prepare_environment():
         args.skip_python_version_check = True
     elif args.use_zluda:
         print('WARNING: ZLUDA works best with SD.Next. Please consider migrating to SD.Next.')
-        backend = "cuda"
+        backend = "zluda"
     elif args.use_ipex:
         backend = "ipex"
         if system == "Windows":
@@ -489,7 +489,7 @@ def prepare_environment():
             if rocm.is_installed:
                 if system == "Windows": # ZLUDA
                     args.use_zluda = True
-                    backend = "cuda"
+                    backend = "zluda"
                 else:
                     backend = "rocm"
                     torch_index_url = os.environ.get(
@@ -539,7 +539,7 @@ def prepare_environment():
     print(f"Version: {tag}")
     print(f"Commit hash: {commit}")
 
-    if args.use_zluda or backend == "rocm":
+    if backend in ("rocm", "zluda",):
         device = None
         try:
             amd_gpus = rocm.get_agents()
@@ -612,7 +612,7 @@ def prepare_environment():
 
     if args.skip_torch_cuda_test:
         print("WARNING: you should not skip torch test unless you want CPU to work.")
-    if args.use_ipex or args.use_directml or args.use_zluda or args.use_cpu_torch:
+    if args.use_ipex or args.use_directml or args.use_cpu_torch:
         args.skip_torch_cuda_test = True
 
     if rocm.is_installed:

modules/shared_init.py

@@ -71,4 +71,5 @@ def initialize():
         from modules.onnx_impl import initialize_onnx
         initialize_onnx()
 
-    initialize_zluda()
+    if devices.backend == "zluda":
+        initialize_zluda()