yankf21
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
mindspore2022/mindspore/_extends/graph_kernel/expanders
History
dayschan 37a48f6aac GraphKernel supports GPU 
1. Update akg submodule
2. Refactor akg_kernel_build, akg_ascend_kernel_build, akg_gpu_kernel_build
3. Add akg_kernel_json_decoder to support converting kernel_json to AnfNode.
4. Add GraphKernel Cost Model. (mindspore/_extends/graph_kernel)
5. Add some GraphKernel passes to GpuSession, move these passes to backend/optimizer/graph_kernel.
6. Add global id for ir files.
7. Fix bug in ConstInputToAttr.
 		2020-09-09 17:46:50 +08:00
..
__init__.py GraphKernel supports GPU 2020-09-09 17:46:50 +08:00
gelu.py GraphKernel supports GPU 2020-09-09 17:46:50 +08:00
layernorm.py GraphKernel supports GPU 2020-09-09 17:46:50 +08:00
softmax.py GraphKernel supports GPU 2020-09-09 17:46:50 +08:00
square.py GraphKernel supports GPU 2020-09-09 17:46:50 +08:00