Skip to content

Commit 7536a14

Browse files
committed
triton init improvements
1 parent 7f0c95b commit 7536a14

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

modules/shared_init.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -71,10 +71,10 @@ def initialize():
7171
from modules.onnx_impl import initialize_onnx
7272
initialize_onnx()
7373

74-
if sys.platform == "win32" and (devices.backend == "zluda" or devices.backend == "rocm"):
74+
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:
7579
from modules.rocm_triton_windows import apply_triton_patches
7680
apply_triton_patches()
77-
78-
if devices.backend == "zluda":
79-
from modules.zluda import initialize_zluda
80-
initialize_zluda()

0 commit comments

Comments
 (0)