revert the back BA from exeprimenents with AB + device_use_mem

This commit is contained in:
myshevts 2021-12-01 10:22:40 +03:00
parent 43266c585e
commit ddbeff3d46

View File

@ -616,7 +616,7 @@ int main(int argc, char* argv[]) {
InferRequestsQueue inferRequestsQueue(exeNetwork, nireq);
if (isFlagSetInCommandLine("use_device_mem")) {
if (device_name.find("GPU") != std::string::npos)
if (device_name.find("GPU") == 0)
::gpu::fillRemoteBlobs(inputFiles, batchSize, app_inputs_info, inferRequestsQueue.requests, exeNetwork);
else if (device_name.find("CPU") == 0)
fillBlobs(inputFiles, batchSize, app_inputs_info, inferRequestsQueue.requests);