Skip to content

GitLab

Commit f014fb96 authored by Umang Yadav's avatar Umang Yadav
Browse files

Fix eliminate_fp8 pass

parent 8734ffa3
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
src/eliminate_fp8.cpp
View file @ f014fb96
...@@ -37,7 +37,8 @@ void eliminate_fp8::apply(module& m) const ...@@ -37,7 +37,8 @@ void eliminate_fp8::apply(module& m) const
{ {
for(auto ins : iterator_for(m)) 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; continue;
migraphx::shape::type_t orig_type = ins->get_shape().type(); migraphx::shape::type_t orig_type = ins->get_shape().type();
std::vector<instruction_ref> orig_inputs = ins->inputs(); std::vector<instruction_ref> orig_inputs = ins->inputs();
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment