[CINN] Optimize parallel compiler and support dumping more compilation information #55590
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
1. Unify FLAGS_cinn_parallel_compile_size and FLAGS_cinn_parallel_compile_thread 2. Add more flags to dump more compile info 3. Support dump lower_func, source_code, ptx and instruction 4. Return more compile info from parallel compiler to graph compiler 5. Refactor graph visualization 6. Refactor parallel compiler's task split
|
你的PR提交成功，感谢你对开源项目的贡献! |
BiynXu
reviewed
Aug 16, 2023
A task corresponds to a fusion group, which is assigned to available threads
BiynXu
reviewed
Aug 18, 2023
Skip test case on op constant
BeingGod
pushed a commit
to BeingGod/Paddle
that referenced
this pull request
Sep 9, 2023
…n information (PaddlePaddle#55590) graph_compiler_util.h/cc： 整合GraphCompiler与ParallelCompiler共同持有的数据结构CompilationStage、CompilationStatus、CompilationContext、CompilationResult Parallel Compiler： 整合数据结构至CompilationContext 支持分阶段编译，通过指定CompilationContext::Stage 添加编译状态信息，包括状态CompilationResult::Status和信息CompilationResult::message 一个Task对应一个fusion_group，每一阶段编译完成后，将编译结果放入CompilationResult数组的对应下标中，删去原Task中内部的局部变量，省去了MergeResult Graph Compiler： 丰富CompilationResult，提供编译各阶段的中间结果 整合数据结构至CompilationContext 添加编译状态信息，包括状态CompilationResult::Status和信息CompilationResult::message 启用单测 其他：相关单测、前端接口、paddle2cinn适配CompilationContext
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
PR types
Others
PR changes
Others
Description
Pcard-73602
graph_compiler_util.h/cc：GraphCompiler与ParallelCompiler共同持有的数据结构CompilationStage、CompilationStatus、CompilationContext、CompilationResultParallel Compiler：CompilationContextCompilationContext::StageCompilationResult::Status和信息CompilationResult::messageTask对应一个fusion_group，每一阶段编译完成后，将编译结果放入CompilationResult数组的对应下标中，删去原Task中内部的局部变量，省去了MergeResultGraph Compiler：CompilationResult，提供编译各阶段的中间结果CompilationContextCompilationResult::Status和信息CompilationResult::messageCompilationContext