diff --git a/mindspore/ccsrc/pre_activate/ascend/enhancer/add_memcpy_async.cc b/mindspore/ccsrc/pre_activate/ascend/enhancer/add_memcpy_async.cc
index 62316f4f6f..51f6732c66 100644
--- a/mindspore/ccsrc/pre_activate/ascend/enhancer/add_memcpy_async.cc
+++ b/mindspore/ccsrc/pre_activate/ascend/enhancer/add_memcpy_async.cc
@@ -25,7 +25,7 @@ namespace opt {
 namespace {
 bool InputIsParameterOrValueNode(const AnfNodePtr &node) {
   MS_EXCEPTION_IF_NULL(node);
-  auto kernel_with_index = AnfAlgo::VisitKernel(node, 0);
+  auto kernel_with_index = AnfAlgo::VisitKernelWithReturnType(node, 0, true);
   return kernel_with_index.first->isa<Parameter>() || kernel_with_index.first->isa<ValueNode>();
 }