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.

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

feature/analysis to support sub-graph for TRT engine #11538

Merged
merged 44 commits into from
Jun 28, 2018

Conversation

Superjomn
Copy link
Contributor

@Superjomn Superjomn commented Jun 18, 2018

  • inference/analysis framework ready for TRT subgraph extraction
  • the whole framework runs smoothly from a fluid model as input, and output some predictions
  • TODO fix some issues in executor, the TensorRTEngineOp seems not executed by executor, but the analysis framework works fine, I will work on it in another PR.

@Superjomn Superjomn changed the title Feature/analysis as debuger feature/analysis as debuger Jun 20, 2018
@Superjomn Superjomn changed the title feature/analysis as debuger feature/analysis to support sub-graph for TRT engine Jun 21, 2018
return true;
}

framework::proto::ProgramDesc ProcessProgramDesc(
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

where is this used?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

cleaned


TEST_F(DFG_Tester, main) {
Analyzer analyser;
analyser.Run(&argument);
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

where does argument come from?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The argument is from DFG_Tester

TEST_F(DFG_Tester, tensorrt_subgraph_node_mark_pass) {
// init
FluidToDataFlowGraphPass pass;
ASSERT_TRUE(pass.Initialize(&argument));
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

why is the argument from?

@Superjomn Superjomn merged commit 5082642 into PaddlePaddle:develop Jun 28, 2018
@Superjomn Superjomn deleted the feature/analysis_as_debuger branch June 28, 2018 01:13
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants