diff --git a/src/gpu/ocl/gen9_global_pooling.cl b/src/gpu/ocl/gen9_global_pooling.cl index 805739589db..69f00acc341 100644 --- 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 index f85c7ffadfd..72bad4a3b7b 100644 --- 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