Skip to content

Commit 26c17c8

Browse files
authored
bugfix: fix core dump of GLM 4.5 when enable MTP. (#336)
1 parent c7e2d65 commit 26c17c8

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

xllm/core/runtime/worker_impl.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -66,6 +66,7 @@ WorkerImpl::WorkerImpl(const ParallelArgs& parallel_args,
6666

6767
device_.set_device();
6868
device_.init_device_context();
69+
threadpool_.schedule([this]() mutable { device_.set_device(); });
6970
general_threadpool_.schedule([this]() mutable { device_.set_device(); });
7071

7172
prepare_stream_ = device_.get_stream_from_pool();

0 commit comments

Comments
 (0)