Update fuse gemm test to use get_compute_fp32_flag() function

ROCm · Jan 3, 2025 · 4e693ba · 4e693ba
1 parent 13159b3
commit 4e693ba
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/test/gpu/fuse_gemm.cpp b/test/gpu/fuse_gemm.cpp
@@ -94,7 +94,10 @@ TEST_CASE(gemm_pointwise_add)
         else
         {
             auto gemm_oper =
-                migraphx::make_op("gpu::gemm", {{"alpha", 1}, {"beta", 1}, {"compute_fp32", 1}});
+                migraphx::make_op("gpu::gemm",
+                                  {{"alpha", 1},
+                                   {"beta", 1},
+                                   {"compute_fp32", migraphx::gpu::get_compute_fp32_flag()}});
             auto add = mm->add_instruction(gemm_oper, a, b, x, output);
             mm->add_return({add});
         }