change gpunum to 3 (#12073)

This commit is contained in:
guozhong wang 2022-07-07 23:15:27 +08:00 committed by GitHub
parent 32937ab7ca
commit ab8c2f6fd8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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",