diff options
author | Won Jeon <won.jeon@arm.com> | 2024-03-13 19:18:08 +0000 |
---|---|---|
committer | Won Jeon <won.jeon@arm.com> | 2024-03-15 21:24:38 +0000 |
commit | e67115ef82bcba0718dcbd75cc8411985001b7cc (patch) | |
tree | ce6896be8d6e25517d913517ef4f5966df787f16 /reference_model | |
parent | 00dc22eea26159661074216688a2c0b9cfb9d709 (diff) | |
download | reference_model-e67115ef82bcba0718dcbd75cc8411985001b7cc.tar.gz |
Add missing FP8 definition and instantiation for FULLY_CONNNECTED op
Signed-off-by: Won Jeon <won.jeon@arm.com>
Change-Id: I7efc03f16136b2b74f8dfef79a6e7e147c93f4ed
Diffstat (limited to 'reference_model')
-rw-r--r-- | reference_model/src/ops/op_factory.cc | 2 | ||||
-rw-r--r-- | reference_model/src/ops/tensor_ops.cc | 2 |
2 files changed, 4 insertions, 0 deletions
diff --git a/reference_model/src/ops/op_factory.cc b/reference_model/src/ops/op_factory.cc index 5dc375d..0f0013c 100644 --- a/reference_model/src/ops/op_factory.cc +++ b/reference_model/src/ops/op_factory.cc @@ -119,6 +119,8 @@ GraphNode* OpFactory::newOp(SubgraphTraverser* sgt, DEF_FACTORY_THREE_TYPE(OpFullyConnected, INT8, INT8, INT32); DEF_FACTORY_THREE_TYPE(OpFullyConnected, INT16, INT8, INT48); DEF_FACTORY_THREE_TYPE(OpFullyConnected, FP64, FP64, FP64); + DEF_FACTORY_THREE_TYPE(OpFullyConnected, FP8E4M3, FP8E4M3, FP16); + DEF_FACTORY_THREE_TYPE(OpFullyConnected, FP8E5M2, FP8E5M2, FP16); break; case Op_MATMUL: DEF_FACTORY_TWO_TYPE_IN_OUT(OpMatMul, FP16, FP16); diff --git a/reference_model/src/ops/tensor_ops.cc b/reference_model/src/ops/tensor_ops.cc index 609265c..7bd249b 100644 --- a/reference_model/src/ops/tensor_ops.cc +++ b/reference_model/src/ops/tensor_ops.cc @@ -2213,6 +2213,8 @@ DEF_INSTANTIATE_THREE_TYPE(OpFullyConnected, INT8, INT4, INT32); DEF_INSTANTIATE_THREE_TYPE(OpFullyConnected, INT8, INT8, INT32); DEF_INSTANTIATE_THREE_TYPE(OpFullyConnected, INT16, INT8, INT48); DEF_INSTANTIATE_THREE_TYPE(OpFullyConnected, FP64, FP64, FP64); +DEF_INSTANTIATE_THREE_TYPE(OpFullyConnected, FP8E4M3, FP8E4M3, FP16); +DEF_INSTANTIATE_THREE_TYPE(OpFullyConnected, FP8E5M2, FP8E5M2, FP16); DEF_INSTANTIATE_TWO_TYPE(OpMatMul, INT8, INT32); DEF_INSTANTIATE_TWO_TYPE(OpMatMul, INT16, INT48); |