From 61acb524ae0331996f5adc081c04938c7ae7b20d Mon Sep 17 00:00:00 2001
From: jumerckx <31353884+jumerckx@users.noreply.github.com>
Date: Mon, 6 Jan 2025 14:25:07 +0100
Subject: [PATCH] fix test

---
 enzyme/test/MLIR/ReverseMode/batched_square.mlir | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/enzyme/test/MLIR/ReverseMode/batched_square.mlir b/enzyme/test/MLIR/ReverseMode/batched_square.mlir
index a69ff6fb96b..86c28670312 100644
--- a/enzyme/test/MLIR/ReverseMode/batched_square.mlir
+++ b/enzyme/test/MLIR/ReverseMode/batched_square.mlir
@@ -13,7 +13,7 @@ module {
 }
 
 // CHECK:  func.func @dsquare(%arg0: f64, %arg1: tensor<2xf64>) -> tensor<2xf64> {
-// CHECK-NEXT:    %0 = call @diffesquare(%arg0, %arg1) : (f64, tensor<2xf64>) -> tensor<2xf64>
+// CHECK-NEXT:    %0 = call @diffe2square(%arg0, %arg1) : (f64, tensor<2xf64>) -> tensor<2xf64>
 // CHECK-NEXT:    return %0 : tensor<2xf64>
 // CHECK-NEXT:  }