fix cube infer op && test=serving

27610f71 · barrierye · 533e16e1 · 27610f71 · 27610f71
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

python/paddle_serving_server/__init__.py python/paddle_serving_server/__init__.py +1 -1

python/paddle_serving_server_gpu/__init__.py python/paddle_serving_server_gpu/__init__.py +1 -1

未找到文件。
--- a/python/paddle_serving_server/__init__.py
+++ b/python/paddle_serving_server/__init__.py
@@ -239,7 +239,7 @@ class Server(object):
            # it from workflow_conf.
            default_engine_names = [
                'general_infer_0', 'general_dist_kv_infer_0',
-                'general_dist_kv_quant_infer'
+                'general_dist_kv_quant_infer_0'
            ]
            engine_name = None
            for node in self.workflow_conf.workflows[0].nodes:

--- a/python/paddle_serving_server_gpu/__init__.py
+++ b/python/paddle_serving_server_gpu/__init__.py
@@ -274,7 +274,7 @@ class Server(object):
            # it from workflow_conf.
            default_engine_names = [
                'general_infer_0', 'general_dist_kv_infer_0',
-                'general_dist_kv_quant_infer'
+                'general_dist_kv_quant_infer_0'
            ]
            engine_name = None
            for node in self.workflow_conf.nodes: