Fix eliminate_fp8 pass

f014fb96 · Umang Yadav · 8734ffa3 · f014fb96
Commit f014fb96 authored Nov 26, 2023 by Umang Yadav
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

src/eliminate_fp8.cpp src/eliminate_fp8.cpp +2 -1

No files found.
--- a/src/eliminate_fp8.cpp
+++ b/src/eliminate_fp8.cpp
@@ -37,7 +37,8 @@ void eliminate_fp8::apply(module& m) const
 {
    for(auto ins : iterator_for(m))
    {
-        if(not contains(op_names, ins->name()))
+        if(not contains(op_names, ins->name()) or
+           ins->get_shape().type() != migraphx::shape::fp8e4m3fnuz_type)
            continue;
        migraphx::shape::type_t orig_type        = ins->get_shape().type();
        std::vector<instruction_ref> orig_inputs = ins->inputs();