Use __enzyme_autodiff binding in square example

joewallwork · joewallwork · commit 3c114978dd66 · 2026-05-07T15:38:18.000+01:00
diff --git a/enzyme/test/Fortran/ForwardMode/allocatableArraySimple.f90 b/enzyme/test/Fortran/ForwardMode/allocatableArraySimple.f90
@@ -58,4 +58,4 @@ program app
 ! CHECK-NEXT: 1
 ! CHECK-NEXT: 0
 ! CHECK-NEXT: 0
-! CHECK-NEXT: 1
+! CHECK-NEXT: 1
diff --git a/enzyme/test/Fortran/ReverseMode/square.f90 b/enzyme/test/Fortran/ReverseMode/square.f90
@@ -12,7 +12,8 @@ real function square( x )
 end module math
 
 program app
-    use math
+    use enzyme, only: __enzyme_autodiff
+    use math, only: square
     implicit none
     external :: __enzyme_autodiff
     real :: x, dx
@@ -27,4 +28,4 @@ program app
 end program app
 
 ! CHECK: 9
-! CHECK-NEXT: 6
+! CHECK-NEXT: 6
