Fix adjust gpu allocation test

src/include/migraphx/instruction.hpp

@@ -26,6 +26,8 @@ struct instruction
 
     void replace(const shape& r);
     
+    void replace(const operation& o);
+
     void recompute_shape();
 
     void clear_arguments();

src/instruction.cpp

@@ -28,6 +28,12 @@ void instruction::replace(const shape& r)
     }
 }
 
+void instruction::replace(const operation& o)
+{
+    op = std::move(o);
+    recompute_shape();
+}
+
 void instruction::recompute_shape() { replace(compute_shape(op, arguments)); }
 
 void instruction::clear_arguments()

src/program.cpp

@@ -63,11 +63,16 @@ static void print_program(const program& p, F print_func)
 
     for(auto ins : iterator_for(p))
     {
-        std::string var_name = "@" + std::to_string(count);
+        std::string var_name;
         if(ins->name() == "@param")
         {
             var_name = any_cast<builtin::param>(ins->get_operator()).parameter;
         }
+        else
+        {
+            var_name = "@" + std::to_string(count);
+            count++;
+        }
         names.emplace(ins, var_name);
 
         // TODO: Use all_of
@@ -78,8 +83,6 @@ static void print_program(const program& p, F print_func)
         }
 
         print_func(ins, names);
-
-        count++;
     }
 }
 

test/gpu/adjust_allocation.cpp

@@ -56,7 +56,7 @@ TEST_CASE(tanh_shape)
         if(ins->name() == "hip::allocate")
         {
             migraphx::shape wrong_s{migraphx::shape::float_type, {3, 2}, {1, 3}};
-            ins->replace(wrong_s);
+            ins->replace(migraphx::gpu::hip_allocate{wrong_s});
         }
     }
     EXPECT(p1 != p2);