add imagenet cpu support

python/examples/pipeline/imagenet/benchmark.py

@@ -32,6 +32,8 @@ def gen_yml(device, gpu_id):
     if device == "gpu":
         config["op"]["imagenet"]["local_service_conf"]["device_type"] = 1
         config["op"]["imagenet"]["local_service_conf"]["devices"] = gpu_id
+    else:
+        config["op"]["imagenet"]["local_service_conf"]["device_type"] = 0
     with open("config2.yml", "w") as fout: 
         yaml.dump(config, fout, default_flow_style=False)
 
@@ -91,7 +93,10 @@ if __name__ == "__main__":
         mode = sys.argv[2] # brpc/  local predictor
         thread = int(sys.argv[3])
         device = sys.argv[4]
+        if device == "gpu":
             gpu_id = sys.argv[5]
+        else:
+            gpu_id = None
         gen_yml(device, gpu_id)
     elif sys.argv[1] == "run":
         mode = sys.argv[2] # http/ rpc

python/examples/pipeline/imagenet/benchmark.sh

 export FLAGS_profile_pipeline=1
 alias python3="python3.7"
 modelname="imagenet"
+use_gpu=0
 gpu_id="0"
 benchmark_config_filename="benchmark_config.yaml"
 
 # HTTP
 ps -ef | grep web_service | awk '{print $2}' | xargs kill -9 
 sleep 3
-python3 benchmark.py yaml local_predictor 1 gpu $gpu_id 
+if [ $use_gpu -eq 1 ]; then
+  python3 benchmark.py yaml local_predictor 1 gpu $gpu_id
+else
+  python3 benchmark.py yaml local_predictor 1 cpu
+fi
 rm -rf profile_log_$modelname
 for thread_num in 1
 do
   for batch_size in 1
   do
-    echo "#----imagenet thread num: $thread_num batch size: $batch_size mode:http ----" >>profile_log_$modelname
+    echo "#----imagenet thread num: $thread_num batch size: $batch_size mode:http use_gpu:$use_gpu----" >>profile_log_$modelname
     rm -rf PipelineServingLogs
     rm -rf cpu_utilization.py
     python3 resnet50_web_service.py >web.log 2>&1 &