revert the back BA from exeprimenents with AB + device_use_mem
This commit is contained in:
parent
43266c585e
commit
ddbeff3d46
@ -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);
|
||||
|
Loading…
Reference in New Issue
Block a user