change parse_onnx to optimize_onnx

ROCm · Jun 3, 2024 · 83e9726 · 83e9726
1 parent c35fb00
commit 83e9726
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 5 deletions.
diff --git a/test/onnx/parse/binary_dyn_brcst_mul_fp8_test.cpp b/test/onnx/parse/binary_dyn_brcst_mul_fp8_test.cpp
@@ -47,7 +47,7 @@ TEST_CASE(binary_dyn_brcst_mul_fp8_test)
 
     migraphx::onnx_options options;
     options.default_dyn_dim_value = {1, 4};
-    auto prog = migraphx::parse_onnx("binary_dyn_brcst_mul_fp8_test.onnx", options);
+    auto prog = read_onnx("binary_dyn_brcst_mul_fp8_test.onnx", options);
 
     EXPECT(p == prog);
 }
diff --git a/test/onnx/parse/gemm_fp8_test.cpp b/test/onnx/parse/gemm_fp8_test.cpp
@@ -32,6 +32,7 @@ TEST_CASE(gemm_fp8_test)
     auto l0    = mm->add_parameter("A", migraphx::shape{migraphx::shape::fp8e4m3fnuz_type, {8, 6}});
     auto l1    = mm->add_parameter("B", migraphx::shape{migraphx::shape::fp8e4m3fnuz_type, {8, 7}});
     auto l2    = mm->add_parameter("C", migraphx::shape{migraphx::shape::fp8e4m3fnuz_type, {6, 1}});
+
     auto alpha = 0.5f;
     auto beta  = 0.8f;
     auto a_l   = mm->add_literal(alpha);

diff --git a/test/onnx/verify/add_fp8_test.cpp b/test/onnx/verify/add_fp8_test.cpp
@@ -29,7 +29,7 @@
 
 TEST_CASE(add_fp8_test)
 {
-    auto p = migraphx::parse_onnx("add_fp8_test.onnx");
+    auto p = optimize_onnx("add_fp8_test.onnx");
     p.compile(migraphx::make_target("ref"));
 
     migraphx::shape s{migraphx::shape::fp8e4m3fnuz_type, {1}};

diff --git a/test/onnx/verify/gemm_fp8_test.cpp b/test/onnx/verify/gemm_fp8_test.cpp
@@ -29,7 +29,7 @@
 
 TEST_CASE(gemm_fp8_test)
 {
-    migraphx::program p = migraphx::parse_onnx("gemm_fp8_test.onnx");
+    migraphx::program p = optimize_onnx("gemm_fp8_test.onnx");
     p.compile(migraphx::make_target("ref"));
 
     migraphx::shape a_shape{migraphx::shape::fp8e4m3fnuz_type, {8, 6}};

diff --git a/test/onnx/verify/shrink_fp8_test.cpp b/test/onnx/verify/shrink_fp8_test.cpp
@@ -29,7 +29,7 @@
 
 TEST_CASE(shrink_fp8_test)
 {
-    migraphx::program p = migraphx::parse_onnx("shrink_fp8_test.onnx");
+    migraphx::program p = optimize_onnx("shrink_fp8_test.onnx");
     p.compile(migraphx::make_target("ref"));
 
     migraphx::shape s{migraphx::shape::fp8e4m3fnuz_type, {3, 3}};

diff --git a/test/onnx/verify/size_fp8_test.cpp b/test/onnx/verify/size_fp8_test.cpp
@@ -29,7 +29,7 @@
 
 TEST_CASE(size_fp8_test)
 {
-    migraphx::program p = migraphx::parse_onnx("size_fp8_test.onnx");
+    migraphx::program p = optimize_onnx("size_fp8_test.onnx");
     p.compile(migraphx::make_target("ref"));
 
     migraphx::shape s{migraphx::shape::fp8e4m3fnuz_type, {2, 5, 3}};