Skip to content

Commit

Permalink
enable brdgmm_dw_conv with non f32 bias(e.g. f16,bf16)
Browse files Browse the repository at this point in the history
  • Loading branch information
liubo-intel committed Dec 27, 2024
1 parent c14134a commit 57abf44
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions src/plugins/intel_cpu/src/nodes/conv.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -980,6 +980,10 @@ void Convolution::createDescriptor(const std::vector<MemoryDescPtr>& inputDesc,
memory::data_type bdt = outDnnlDesc.get_data_type();
#else
memory::data_type bdt = memory::data_type::f32;
// brdgmm_dw_conv supports only bia_type the same as src_type or dst_type
auto out_dt = outDnnlDesc.get_data_type();
if (!canBeExecutedInInt8() && isDepthWise() && out_dt != memory::data_type::f32)
bdt = out_dt;
#endif
biasDnnlDesc =
dnnl::memory::desc(DnnlExtensionUtils::convertToDnnlDims(expectedBiasDims), bdt, memory::format_tag::any);
Expand Down

0 comments on commit 57abf44

Please sign in to comment.