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