We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 7f0c95b commit 7536a14Copy full SHA for 7536a14
modules/shared_init.py
@@ -71,10 +71,10 @@ def initialize():
71
from modules.onnx_impl import initialize_onnx
72
initialize_onnx()
73
74
- if sys.platform == "win32" and (devices.backend == "zluda" or devices.backend == "rocm"):
+ if devices.backend == "zluda":
75
+ from modules.zluda import initialize_zluda
76
+ initialize_zluda()
77
+
78
+ if sys.platform == "win32" and torch.cuda.is_available() and torch.version.hip is not None:
79
from modules.rocm_triton_windows import apply_triton_patches
80
apply_triton_patches()
-
- if devices.backend == "zluda":
- from modules.zluda import initialize_zluda
- initialize_zluda()
0 commit comments