From ede2dfc7f8df8c0eb0e0945851dbd8ec4666ac5c Mon Sep 17 00:00:00 2001 From: alexey-varyzgin Date: Fri, 1 Apr 2022 12:40:53 +0300 Subject: [PATCH] [BF16] ref_pooling_fwd was added for bf16 --- src/cpu/cpu_pooling_list.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/cpu/cpu_pooling_list.cpp b/src/cpu/cpu_pooling_list.cpp index 8f0d7de85b6..96e4dc3aca1 100644 --- a/src/cpu/cpu_pooling_list.cpp +++ b/src/cpu/cpu_pooling_list.cpp @@ -84,8 +84,8 @@ const impl_list_item_t impl_list[] = { REG_POOLING_P_BWD(CPU_INSTANCE(nhwc_pooling_bwd_t, f32)) #endif REG_POOLING_P_FWD(CPU_INSTANCE(ref_pooling_fwd_t, f32, f32, f32)) -#ifdef ENABLE_UNUSED_PRIM REG_POOLING_P_FWD(CPU_INSTANCE(ref_pooling_fwd_t, bf16, bf16, f32)) +#ifdef ENABLE_UNUSED_PRIM REG_POOLING_P_BWD(CPU_INSTANCE(ref_pooling_bwd_t, f32)) REG_POOLING_P_BWD(CPU_INSTANCE(ref_pooling_bwd_t, bf16)) #endif