fp16 change default tolerances

f73162a7 · charlie · e5cd8b6b · f73162a7
Commit f73162a7 authored Oct 10, 2023 by charlie
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 0 deletions

src/driver/main.cpp src/driver/main.cpp +6 -0

No files found.
--- a/src/driver/main.cpp
+++ b/src/driver/main.cpp
@@ -542,6 +542,12 @@ struct verify : command<verify>
    void parse(argument_parser& ap)
    {
        c.parse(ap);
+        //TODO remove this and make the driver able to figure out datatype most used in the model
+        // then set the tolerances appropriately
+        if(c.to_fp16)
+        {
+            tols = migraphx::verify::tolerance{8e-2, 4e-2, 4e-2};
+        }
        ap(tols.rms_tol, {"--rms-tol"}, ap.help("Tolerance for the RMS error (Default: 0.001)"));
        ap(tols.atol,
           {"--atol"},