src: gpu: ocl: pooling: add f16 support to bwd ocl global pooling

oneapi-src · Apr 8, 2024 · e4737d9 · e4737d9
1 parent c8ad13c
commit e4737d9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/gpu/ocl/gen9_global_pooling.cl b/src/gpu/ocl/gen9_global_pooling.cl
@@ -1,5 +1,5 @@
 /*******************************************************************************
-* Copyright 2021-2023 Intel Corporation
+* Copyright 2021-2024 Intel Corporation
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
@@ -81,7 +81,7 @@ __kernel void gen9_global_pooling_fwd(
 
 #if IS_BWD
 
-#if DT_BF16
+#if DT_BF16 || DT_F16
 #define DST_BLOCK_WRITE(dst, val) \
     BLOCK_WRITE((__global ushort *)(dst), as_ushort(val))
 #endif // DT_BF16

diff --git a/tests/benchdnn/inputs/pool/test_pool_gpu b/tests/benchdnn/inputs/pool/test_pool_gpu
@@ -89,7 +89,7 @@
 
 # Test global pooling
 --reset
---dt=f32,bf16,f64
+--dt=f32,f16,bf16,f64
 --dir=BWD_D
 --tag=axb,abx,aBx16b,aBx32b
 --alg=max,avg_np,avg_p