cpu: dw post-op: allow scales for int8 conv

oneapi-src · Jul 26, 2021 · d606546 · d606546
1 parent 050bc30
commit d606546
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/cpu/dw_convolution_utils.hpp b/src/cpu/dw_convolution_utils.hpp
@@ -42,9 +42,7 @@ inline status_t get_depthwise_conv_desc(convolution_desc_t &cd_dw,
     // Create new attributes with scales from depthwise post-op and copy
     // post-ops after depthwise post-op.
     auto &dw_po = attr_1x1.post_ops_.entry_[dw_po_index].depthwise_conv;
-    if (utils::one_of(
-                dw_po.dst_dt, data_type::u8, data_type::s8, data_type::s32)
-            && dw_po.count) {
+    if (dw_po.wei_dt == data_type::s8 && dw_po.count) {
         CHECK(attr_dw.output_scales_.set(
                 dw_po.count, dw_po.mask, dw_po.scales));
     }