diff --git a/caffe2/opt/fakefp16_transform.cc b/caffe2/opt/fakefp16_transform.cc index 4db67410e41..10c54ef4734 100644 --- a/caffe2/opt/fakefp16_transform.cc +++ b/caffe2/opt/fakefp16_transform.cc @@ -43,6 +43,7 @@ std::unordered_map getFakeFp16OpMapping( "SparseLengthsWeightedSumFused8BitRowwiseFakeFP16NNPI"}, {"SparseLengthsMeanFused8BitRowwise", "SparseLengthsMeanFused8BitRowwiseFakeFP16AccFP16"}, + {"MatMul", "BatchMatMulFP16Acc32Fake"}, {"BatchMatMul", "BatchMatMulFP16Acc32Fake"}, {"Sigmoid", "SigmoidFakeFp16"}, {"SpatialBN", "SpatialBNFakeFp16NNPI"}, @@ -60,6 +61,7 @@ std::unordered_map getFakeFp16OpMapping( fake_fp16_op_conversion_map["FC"] = "Fp16FCAcc16NNPI"; fake_fp16_op_conversion_map["FbFCPacked"] = "Fp16FCAcc16NNPI"; fake_fp16_op_conversion_map["BatchMatMul"] = "BatchMatMulFP16Acc16Fake"; + fake_fp16_op_conversion_map["MatMul"] = "BatchMatMulFP16Acc16Fake"; } if (use_nnpi) { fake_fp16_op_conversion_map["Sigmoid"] = "SigmoidFakeFp16NNPI";