diff --git a/lite/core/memory.h b/lite/core/memory.h index 77589a536521506db3debee466e45719a7d796ce..bc382d8c36d87735bbd3028ff06b05f3394bf377 100644 --- a/lite/core/memory.h +++ b/lite/core/memory.h @@ -76,6 +76,11 @@ void CopySync(void* dst, const void* src, size_t size, IoDirection dir) { TargetWrapperCL::MemcpySync(dst, src, size, dir); break; #endif // LITE_WITH_OPENCL +#ifdef LITE_WITH_MLU + case TARGET(kMLU): + TargetWrapperMlu::MemcpySync(dst, src, size, dir); + break; +#endif #ifdef LITE_WITH_FPGA case TARGET(kFPGA): TargetWrapper::MemcpySync(dst, src, size, dir);