[CPU] [ARM] FullyConnected: int8 support: mapping

eshoguli · Aug 13, 2024 · d9b4f89 · d9b4f89
1 parent 9c69960
commit d9b4f89
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/plugins/intel_cpu/src/nodes/executors/acl/acl_lowp_fullyconnected.cpp b/src/plugins/intel_cpu/src/nodes/executors/acl/acl_lowp_fullyconnected.cpp
@@ -71,9 +71,13 @@ ACLLowpFullyConnectedExecutor::ACLLowpFullyConnectedExecutor(const FCAttrs &attr
 
 bool ACLLowpFullyConnectedExecutor::supports(const FCConfig &config) {
     const auto src0 = srcType(config);
-    //const auto src1 = weiType(config);
+    const auto src1 = weiType(config);
     //const auto dst = dstType(config);
 
+    if ((src0 == element::f16) || (src1 == element::f16)) {
+        std::cout << "ACLLowpFullyConnectedExecutor::supports" << std::endl;
+    }
+
     // TODO: check precisions
     VERIFY(one_of(src0, ov::element::i8, ov::element::u8), UNSUPPORTED_SRC_PRECISIONS);
     //VERIFY(postOpsNumbers(config) == 0, UNSUPPORTED_NUMBER_OF_POSTOPS);

diff --git a/src/plugins/intel_cpu/src/nodes/executors/fullyconnected_implementations.cpp b/src/plugins/intel_cpu/src/nodes/executors/fullyconnected_implementations.cpp
@@ -91,7 +91,7 @@ static const TypeMapping aclLowpFCTypeMapping {
     {{_i8, _i8, _any, _f32},                 pt(just<i8>(), just<i8>(), just<i32>(), just<f32>())},
     //{{_i8, _i8, _any, _i32},                 pt(just<i8>(), just<i8>(), just<i32>(), just<i32>())},
     //{{_u8, _u8, _any, _i32},                 pt(just<u8>(), just<u8>(), bypass(), just<i32>())},
-    {{_any, _any, _any, _any},               pt(just<f32>(), just<f32>(), just<f32>(), just<f32>())}
+    //{{_any, _any, _any, _any},               pt(just<f32>(), just<f32>(), just<f32>(), just<f32>())}
 };
 
 static const MappingNotation dnnlConvolutionMappingNotation {