We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c7e2d65 commit 26c17c8Copy full SHA for 26c17c8
xllm/core/runtime/worker_impl.cpp
@@ -66,6 +66,7 @@ WorkerImpl::WorkerImpl(const ParallelArgs& parallel_args,
66
67
device_.set_device();
68
device_.init_device_context();
69
+ threadpool_.schedule([this]() mutable { device_.set_device(); });
70
general_threadpool_.schedule([this]() mutable { device_.set_device(); });
71
72
prepare_stream_ = device_.get_stream_from_pool();
0 commit comments