Add xpu2 compiler #37254
Merged
Add xpu2 compiler #37254
Conversation
This file contains hidden or 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
| Thanks for your contribution! |
Liu-xiandong force-pushed the Add_xpu_compiler branch from ea413a5 to 8a9cdad Compare | Sorry to inform you that d473766's CIs have passed for more than 7 days. To prevent PR conflicts, you need to re-run all CIs manually. |
| Sorry to inform you that bdaa02c's CIs have passed for more than 7 days. To prevent PR conflicts, you need to re-run all CIs manually. |
xingfeng01 reviewed Jan 24, 2022
xingfeng01 reviewed Jan 27, 2022
xingfeng01 previously approved these changes Jan 28, 2022
lanxianghit previously approved these changes Jan 28, 2022
Liu-xiandong dismissed stale reviews from lanxianghit and xingfeng01 via b55d391 
wanghuancoder approved these changes Jan 29, 2022
xingfeng01 approved these changes Jan 29, 2022
lanxianghit approved these changes Jan 29, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
5 participants
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
New features
PR changes
Others
Describe
This PR adds the source code compilation mode of XPU2, and the compilation commands are as follows:
Its realization mainly involves the following aspects:
1. Support source code compilation mode of XPU2 code:
Mainly related to the modification of cmake, the main files involved are: cmake/generic.cmake cmake/operators.cmake cmake/xpu_kp.cmake
The specific calling process is as follows:
2. Compatible operator mapping mode and source code compilation mode
After adding the source code compilation mode, there are two XPU kernels for the same OP.
In order to be compatible with the two modes, three aspects of work have been done.
Added a new library_type to distinguish the original mode.
The main files involved are: paddle/fluid/framework/library_type.h
In order to dynamically select the kernel at runtime, the flag and the list involved in the KP mode are added.
The main files involved are: paddle/fluid/platform/flags.cc paddle/fluid/platform/device/xpu/xpu_op_kpfirst_list.h paddle/fluid/platform/device/xpu/xpu_op_list.cc
Increase specific runtime control methods
The main files involved are:
paddle/fluid/framework/operator.cc (static graph)
paddle/fluid/imperative/prepared_operator.cc (dynamic graph)
The control graph of runtime is as follows:
