[SYCLomatic] Refine help function async_reduction #309

Signed-off-by: Wang, Hao3 <[email protected]>
oneapi-src · Oct 21, 2022 · c093888 · c093888
1 parent 063bc30
commit c093888
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 8 deletions.
diff --git a/clang/runtime/dpct-rt/include/dnnl_utils.hpp.inc b/clang/runtime/dpct-rt/include/dnnl_utils.hpp.inc
@@ -2965,10 +2965,13 @@ sycl::event engine_ext::async_reduction(reduction_op op, float alpha,
   auto execution_args = new std::unordered_map<int, ::dnnl::memory>{
       {DNNL_ARG_SRC, ::dnnl::memory(src_desc.get_desc(), _eng, src)}};
 
-  auto e =
-      execute_primitive(primitive, execution_args,
-                        {{alpha, beta, DNNL_ARG_DST, dst_desc, dst}}, {cache});
-  return e;
+  if (cache) {
+    return execute_primitive(primitive, execution_args,
+                             {{alpha, beta, DNNL_ARG_DST, dst_desc, dst}},
+                             {cache});
+  }
+  return execute_primitive(primitive, execution_args,
+                           {{alpha, beta, DNNL_ARG_DST, dst_desc, dst}});
 }
 // DPCT_LABEL_END
 

diff --git a/clang/test/dpct/helper_files_ref/include/dnnl_utils.hpp b/clang/test/dpct/helper_files_ref/include/dnnl_utils.hpp
@@ -2624,10 +2624,13 @@ sycl::event engine_ext::async_reduction(reduction_op op, float alpha,
   auto execution_args = new std::unordered_map<int, ::dnnl::memory>{
       {DNNL_ARG_SRC, ::dnnl::memory(src_desc.get_desc(), _eng, src)}};
 
-  auto e =
-      execute_primitive(primitive, execution_args,
-                        {{alpha, beta, DNNL_ARG_DST, dst_desc, dst}}, {cache});
-  return e;
+  if (cache) {
+    return execute_primitive(primitive, execution_args,
+                             {{alpha, beta, DNNL_ARG_DST, dst_desc, dst}},
+                             {cache});
+  }
+  return execute_primitive(primitive, execution_args,
+                           {{alpha, beta, DNNL_ARG_DST, dst_desc, dst}});
 }
 
 sycl::event engine_ext::async_activation_forward(activation_desc &desc, float alpha,