| 21 | |
| 22 | template<typename Context> |
| 23 | std::unique_ptr<ep::primitive::Memcpy> NewMemcpyPrimitive(Context* ctx) { |
| 24 | return ep::primitive::NewPrimitive<ep::primitive::MemcpyFactory>( |
| 25 | ctx->device_type(), ep::primitive::MemcpyKind::kDtoD); |
| 26 | } |
| 27 | template<typename T> |
| 28 | class CpuModeKernel final : public user_op::OpKernel { |
| 29 | public: |