ad fp8 warnings (#2531)

dfc18d6c · Umang Yadav · GitHub · 6a72e8fc · dfc18d6c
Unverified Commit dfc18d6c authored Dec 07, 2023 by Umang Yadav Committed by GitHub Dec 07, 2023
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: