enable fp16 convert tests

This commit is contained in:
Aleksandr Voron committed 2023-07-03 17:00:48 +02:00
1 parent 3db3d4294e
commit 037af672ce
2 files changed
+3 -2

No files matched your search

@@ -45,7 +45,7 @@ public:
InferenceEngine::Precision::FP16,
InferenceEngine::Precision::I32,
InferenceEngine::Precision::FP32)) {
DEBUG_LOG("NECopy does not support source precision: ", convertParams.srcPrc.name());
DEBUG_LOG("NECast does not support source precision: ", convertParams.srcPrc.name());
return false;
}
if ((convertParams.srcPrc == InferenceEngine::Precision::I8 &&
@@ -80,7 +80,7 @@ public:
!one_of(convertParams.dstPrc, InferenceEngine::Precision::BF16,
InferenceEngine::Precision::FP16,
InferenceEngine::Precision::I32))) {
DEBUG_LOG("NECopy does not support passed combination of source and destination precisions. ",
DEBUG_LOG("NECast does not support passed combination of source and destination precisions. ",
"source precision: ", convertParams.srcPrc.name(), " destination precsion: " , convertParams.dstPrc.name());
return false;
}
@@ -167,6 +167,7 @@ const std::vector<Precision>& precisions() {
Precision::U8,
Precision::I8,
Precision::I32,
Precision::FP16,
Precision::FP32,
Precision::BF16
};