Commit Graph

3 Commits

Author SHA1 Message Date
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
duxiutao 793737ab62 add primitive operator to test_lamb 2020-07-02 16:03:01 +08:00
duxiutao 1e43c609e0 Add test case and fix two bugs
1. add case to guard precision
2. fix a shape bug
3. fix a funcGraph bug
2020-07-02 12:52:15 +08:00