提交 9473ab6c 编写于 作者: T tensor-tang

fix splite and disable gencode

上级 0e250d51
...@@ -18,11 +18,11 @@ if (NOT LITE_WITH_LIGHT_WEIGHT_FRAMEWORK) ...@@ -18,11 +18,11 @@ if (NOT LITE_WITH_LIGHT_WEIGHT_FRAMEWORK)
DEPS scope_lite op_lite kernel_lite paddle_infer_gencode DEPS scope_lite op_lite kernel_lite paddle_infer_gencode
) )
lite_cc_test(test_generated_code SRCS generated_code_test.cc DEPS __generated_code__ # lite_cc_test(test_generated_code SRCS generated_code_test.cc DEPS __generated_code__
${ops_lite} ${host_kernels} # ${ops_lite} ${host_kernels}
X86_DEPS ${x86_kernels} # X86_DEPS ${x86_kernels}
) # )
add_dependencies(__generated_code__ test_gen_code_lite) # add_dependencies(__generated_code__ test_gen_code_lite)
endif() endif()
...@@ -37,7 +37,7 @@ bool SplitOp::InferShape() const { ...@@ -37,7 +37,7 @@ bool SplitOp::InferShape() const {
const auto &sections = param_.sections; const auto &sections = param_.sections;
const int outs_number = outs.size(); const int outs_number = outs.size();
std::vector<lite::DDimLite> outs_dims; std::vector<lite::DDimHvy> outs_dims;
outs_dims.reserve(outs_number); outs_dims.reserve(outs_number);
if (num > 0) { if (num > 0) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册