From d084ca53459e423fffc66271b94dcf2402cf2946 Mon Sep 17 00:00:00 2001 From: MRXLT Date: Wed, 8 Jul 2020 10:52:04 +0800 Subject: [PATCH] fix ci --- python/examples/criteo_ctr_with_cube/benchmark_cube.sh | 4 ++-- python/examples/criteo_ctr_with_cube/test_server_gpu.py | 6 +++++- .../criteo_ctr_with_cube/test_server_gpu.py | 6 +++++- 3 files changed, 12 insertions(+), 4 deletions(-) diff --git a/python/examples/criteo_ctr_with_cube/benchmark_cube.sh b/python/examples/criteo_ctr_with_cube/benchmark_cube.sh index aaf1dcce..d0964a4a 100755 --- a/python/examples/criteo_ctr_with_cube/benchmark_cube.sh +++ b/python/examples/criteo_ctr_with_cube/benchmark_cube.sh @@ -14,9 +14,9 @@ sh cube_prepare.sh & cp ../../../build_server/core/cube/cube-api/cube-cli . python gen_key.py -for thread_num in 1 +for thread_num in 1 4 16 32 do -for batch_size in 1 +for batch_size in 1000 do ./cube-cli -config_file ./cube/conf/cube.conf -keys key -dict test_dict -thread_num $thread_num --batch $batch_size > profile 2>&1 echo "batch size : $batch_size" diff --git a/python/examples/criteo_ctr_with_cube/test_server_gpu.py b/python/examples/criteo_ctr_with_cube/test_server_gpu.py index 382be99b..33f74f91 100755 --- a/python/examples/criteo_ctr_with_cube/test_server_gpu.py +++ b/python/examples/criteo_ctr_with_cube/test_server_gpu.py @@ -33,5 +33,9 @@ server = Server() server.set_op_sequence(op_seq_maker.get_op_sequence()) server.set_num_threads(4) server.load_model_config(sys.argv[1]) -server.prepare_server(workdir="work_dir1", port=9292, device="cpu") +server.prepare_server( + workdir="work_dir1", + port=9292, + device="cpu", + cube_conf="./cube/conf/cube.conf") server.run_server() diff --git a/python/examples/grpc_impl_example/criteo_ctr_with_cube/test_server_gpu.py b/python/examples/grpc_impl_example/criteo_ctr_with_cube/test_server_gpu.py index 38e1bf82..343ded24 100755 --- a/python/examples/grpc_impl_example/criteo_ctr_with_cube/test_server_gpu.py +++ b/python/examples/grpc_impl_example/criteo_ctr_with_cube/test_server_gpu.py @@ -33,5 +33,9 @@ server = Server() server.set_op_sequence(op_seq_maker.get_op_sequence()) server.set_num_threads(4) server.load_model_config(sys.argv[1], sys.argv[2]) -server.prepare_server(workdir="work_dir1", port=9292, device="cpu") +server.prepare_server( + workdir="work_dir1", + port=9292, + device="cpu", + cube_conf="./cube/conf/cube.conf") server.run_server() -- GitLab