gpu: ocl: post_ops: fixup typo

oneapi-src · May 31, 2023 · ad3c62f · ad3c62f
1 parent 7fa3b6f
commit ad3c62f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/gpu/ocl/ocl_post_ops.h b/src/gpu/ocl/ocl_post_ops.h
@@ -68,8 +68,9 @@ float fwd_Xnary(unsigned algorithm, float x, float y, float alpha, float beta,
         unroll_for(typeof(out_len + 0) idx = 0; idx < out_len; ++idx) { \
             const int arg0_idx = arg0_len == 1 ? 0 : idx; \
             const int arg1_idx = arg1_len == 1 ? 0 : idx; \
-            res_ptr[idx] = fwd_Xnary(algorithm, convert_float(arg0_ptr[idx]), \
-                    convert_float(arg1_ptr[idx]), alpha, beta, scale); \
+            res_ptr[arg0_len == 1 && arg1_len == 1 ? 0 : idx] = fwd_Xnary( \
+                    po_kind, algorithm, convert_float(arg0_ptr[arg0_idx]), \
+                    convert_float(arg1_ptr[arg1_idx]), alpha, beta, scale); \
         } \
     }