| 23 | } // namespace |
| 24 | |
| 25 | codegen::AutoKernelFunc* codegen::GenCode(KernelPack::KernType kern_type) { |
| 26 | static AllMLIRKernel all_kernel; |
| 27 | if (all_kernel.inner_map.find(kern_type) != all_kernel.inner_map.end()) { |
| 28 | return all_kernel.inner_map[kern_type].get(); |
| 29 | } else { |
| 30 | return nullptr; |
| 31 | } |
| 32 | } |