diff --git a/src/native/cambricon/runtime_.h b/src/native/cambricon/runtime_.h index c4f4112..4e89928 100644 --- a/src/native/cambricon/runtime_.h +++ b/src/native/cambricon/runtime_.h @@ -5,7 +5,6 @@ #include #include -#include #include "native/cambricon/device_.h" #include "runtime.h" @@ -67,12 +66,7 @@ struct Runtime static constexpr auto MemcpyAsync = [](void* dst, const void* src, std::size_t size, auto kind, Stream stream) { - if constexpr (std::is_invocable_v) { - return cnrtMemcpyAsync(dst, const_cast(src), size, kind, stream); - } else { - return cnrtMemcpyAsync(dst, const_cast(src), size, stream, kind); - } + return cnrtMemcpyAsync_V2(dst, const_cast(src), size, stream, kind); }; static constexpr auto kMemcpyHostToHost = cnrtMemcpyHostToHost;