未验证 提交 9382159d 编写于 作者: H Hui Zhang 提交者: GitHub

Modify PE Engine thread from 2 into 1 in JitLayer (#45356)

上级 04c15e79
...@@ -31,7 +31,7 @@ static ExecutionStrategy GetExecutionStrategy(const platform::Place &place) { ...@@ -31,7 +31,7 @@ static ExecutionStrategy GetExecutionStrategy(const platform::Place &place) {
auto device_type = platform::Place2DeviceType(place); auto device_type = platform::Place2DeviceType(place);
switch (device_type) { switch (device_type) {
case platform::DeviceType::CPU: { case platform::DeviceType::CPU: {
execution_strategy.num_threads_ = 2; execution_strategy.num_threads_ = 1;
break; break;
} }
case platform::DeviceType::CUDA: { case platform::DeviceType::CUDA: {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册