Skip to content

Commit 13343af

Browse files
authored
disable check_env in multiprocess on dlinfer devices (#3914)
1 parent f86964f commit 13343af

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

lmdeploy/pytorch/engine/engine_checker.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,7 @@ def handle(self):
109109
return
110110

111111
current_proc = mp.current_process()
112-
if not current_proc.daemon:
112+
if not current_proc.daemon and self.engine_config.device_type == 'cuda':
113113
mp_ctx = mp.get_context('spawn')
114114
with ProcessPoolExecutor(mp_context=mp_ctx) as executor:
115115
try:

0 commit comments

Comments
 (0)