change gpunum to 3 (#12073)
This commit is contained in:
parent
32937ab7ca
commit
ab8c2f6fd8
@ -146,7 +146,7 @@ void AutoSchedule::init(const ScheduleContext::Ptr& sContext) {
|
||||
}
|
||||
|
||||
// remove CPU from default candidate list for Cumulative Throughput mode
|
||||
if (GPUNums >= 2 && CPUNums > 0) {
|
||||
if (GPUNums >= 3 && CPUNums > 0) {
|
||||
validDevices.erase(itCPUDevice);
|
||||
LOG_INFO_TAG("GPUNums:%d, remove CPU from default candidate list for "
|
||||
"CUMULATIVE_THROUGHPUT",
|
||||
|
Loading…
Reference in New Issue
Block a user