gpu: generic: implement SYCL reduction (#2243)

oneapi-src · Jan 16, 2025 · ea6c0b7 · ea6c0b7
1 parent 5cd6a3b
commit ea6c0b7
Show file tree

Hide file tree

Showing 11 changed files with 1,385 additions and 7 deletions.
diff --git a/src/common/memory_tracking.hpp b/src/common/memory_tracking.hpp
@@ -280,6 +280,7 @@ enum {
     key_reducer_space_bctx,
     key_reduction,
     key_reduction_1,
+    key_reduction_out,
     key_reorder_cross_space,
     key_reorder_space,
     key_reorder_src_scales,

diff --git a/src/gpu/generic/sycl/README.md b/src/gpu/generic/sycl/README.md
@@ -174,3 +174,8 @@ The implementation supports both forward and backward propagations.
 
 * Supported formats: plain formats with up to 7 dimensions
 * Supported data types: `f32`, `bf16`, `f16`, `s8`, `u8`
+
+## Reduction
+
+* Supported formats: plain formats with up to 6 dimensions
+* Supported data types: `f32`, `bf16`, `f16`, `s8`, `u8`