| 135 | */ |
| 136 | |
| 137 | PD_KernelArgsDef* PD_KernelGetArgsDef(PD_Kernel* kernel, PD_Status* status) { |
| 138 | if (status) { |
| 139 | if (!kernel) { |
| 140 | *status = C_FAILED; |
| 141 | return nullptr; |
| 142 | } |
| 143 | *status = C_SUCCESS; |
| 144 | } |
| 145 | auto cc_kernel = reinterpret_cast<phi::Kernel*>(kernel); |
| 146 | return reinterpret_cast<PD_KernelArgsDef*>( |
| 147 | const_cast<phi::KernelArgsDef*>(&cc_kernel->args_def())); |
| 148 | } |
| 149 | |
| 150 | PD_REGISTER_CAPI(kernel_factory); |