ad fp8 warnings

4a5dcb97 · Umang Yadav · 6a72e8fc · 4a5dcb97
Commit 4a5dcb97 authored Dec 07, 2023 by Umang Yadav
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

src/onnx/onnx_parser.cpp src/onnx/onnx_parser.cpp +5 -1

No files found.
--- a/src/onnx/onnx_parser.cpp
+++ b/src/onnx/onnx_parser.cpp
@@ -625,7 +625,11 @@ shape::type_t get_type(int dtype)
    case 11: return shape::double_type;
    case 12: return shape::uint32_type;
    case 13: return shape::uint64_type;
-    case 18: return shape::fp8e4m3fnuz_type;
+    case 18: {
+        std::cout << "[Warning] : MIGraphX has BETA support for FP8. Using FP8 may result in "
+                     "incorrect final outputs\n";
+        return shape::fp8e4m3fnuz_type;
+    }
    case 14:
    case 15:
    case 16: