Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Support control flow for static build [Step 1: support subgraph] #56185

Merged
merged 13 commits into from
Aug 17, 2023
Merged

Support control flow for static build [Step 1: support subgraph] #56185

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
13 commits
Select commit Hold shift + click to select a range
ed59157
remove execution_config.used_for_control_flow_op
AndSonder Aug 11, 2023
af02b65
update
AndSonder Aug 11, 2023
fba2e52
update
AndSonder Aug 11, 2023
6c6e7c0
open static build flag
AndSonder Aug 12, 2023
598bac8
close static build flag
AndSonder Aug 12, 2023
6202068
open static build flag
AndSonder Aug 12, 2023
6d4b25a
Merge branch 'develop' of https://github.com/PaddlePaddle/Paddle into…
AndSonder Aug 13, 2023
bc413f6
add searchsorted to analyze dtype list
AndSonder Aug 14, 2023
2305ec6
Merge branch 'develop' of https://github.com/PaddlePaddle/Paddle into…
AndSonder Aug 15, 2023
5fc5f51
Merge branch 'develop' of https://github.com/PaddlePaddle/Paddle into…
AndSonder Aug 15, 2023
55f99dc
recover and add test_searchsorted_op to static build list
AndSonder Aug 15, 2023
0d70a2d
Update CMakeLists.txt
AndSonder Aug 16, 2023
ba2f9fd
Update CMakeLists.txt
AndSonder Aug 16, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 7 additions & 0 deletions paddle/fluid/framework/new_executor/interpreter/static_build.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -474,6 +474,13 @@ void FakeInitializeOutputsForFunctionKernel(
? DataType::INT64
: in_dtype;
}
} else if (op_type == "searchsorted") {
bool out_int32 = op.Attr<bool>("out_int32");
if (out_int32) {
dtype = DataType::INT32;
} else {
dtype = DataType::INT64;
}
} else {
VLOG(4) << "Get dtype result from InferMeta";
RuntimeInferShapeContext infer_shape_ctx(op, runtime_ctx);
Expand Down
1 change: 0 additions & 1 deletion paddle/fluid/framework/new_executor/program_interpreter.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,6 @@ ProgramInterpreter::ProgramInterpreter(const platform::Place& place,

static_build_ = FLAGS_new_executor_static_build &&
!FLAGS_new_executor_use_cuda_graph &&
!execution_config.used_for_control_flow_op &&
interpreter::BlockCanBeStaticBuilt(block);

exception_notifier_ = main_thread_blocker_.RegisterEvent(kExceptionCaught);
Expand Down
2 changes: 2 additions & 0 deletions test/legacy_test/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1272,6 +1272,7 @@ set(STATIC_BUILD_TESTS
test_adamw_op
test_arg_min_max_op
test_assign_pos_op
test_bucketize_api
test_bincount_op
test_c_embedding_op
test_decayed_adagrad_op
Expand Down Expand Up @@ -1302,6 +1303,7 @@ set(STATIC_BUILD_TESTS
test_prune_gate_by_capacity_op
test_random_routing_op
test_reduce_op
test_searchsorted_op
test_segment_ops
test_sparse_momentum_op
test_sgd_op_bf16
Expand Down