From fe928dfff541142a713288238df0c2d038db41f7 Mon Sep 17 00:00:00 2001 From: Stanley Seibert Date: Mon, 31 Jan 2022 10:50:16 -0800 Subject: [PATCH] fix profile tests for MLIR 14 --- mlir_graphblas/tests/test_profile.py | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/mlir_graphblas/tests/test_profile.py b/mlir_graphblas/tests/test_profile.py index ac276a82..a7d7d959 100644 --- a/mlir_graphblas/tests/test_profile.py +++ b/mlir_graphblas/tests/test_profile.py @@ -41,12 +41,12 @@ def test_profile_single_return( function_name = f"single_value_slow_mul_{int(uuid.uuid4())}" mlir_text = f""" func @{function_name}(%input_val: i64) -> i64 {{ - %ci0 = constant 0 : i64 - %c0 = constant 0 : index - %c1 = constant 1 : index - %num_iterations = constant {num_iterations} : index + %ci0 = arith.constant 0 : i64 + %c0 = arith.constant 0 : index + %c1 = arith.constant 1 : index + %num_iterations = arith.constant {num_iterations} : index %answer = scf.for %i = %c0 to %num_iterations step %c1 iter_args(%current_sum=%ci0) -> (i64) {{ - %new_sum = addi %current_sum, %input_val : i64 + %new_sum = arith.addi %current_sum, %input_val : i64 scf.yield %new_sum : i64 }} return %answer : i64 @@ -89,12 +89,12 @@ def test_profile_multiple_returns( function_name = f"multivalue_slow_mul_{int(uuid.uuid4())}" mlir_text = f""" func @{function_name}(%input_val: i64) -> (i64, i64) {{ - %ci0 = constant 0 : i64 - %c0 = constant 0 : index - %c1 = constant 1 : index - %num_iterations = constant {num_iterations} : index + %ci0 = arith.constant 0 : i64 + %c0 = arith.constant 0 : index + %c1 = arith.constant 1 : index + %num_iterations = arith.constant {num_iterations} : index %answer = scf.for %i = %c0 to %num_iterations step %c1 iter_args(%current_sum=%ci0) -> (i64) {{ - %new_sum = addi %current_sum, %input_val : i64 + %new_sum = arith.addi %current_sum, %input_val : i64 scf.yield %new_sum : i64 }} return %input_val, %answer : i64, i64