Update allocation tests for Julia version compatibility

Signed-off-by: Hong Ge <3279477+yebai@users.noreply.github.com>

Author: yebai

src/test_utils.jl

@@ -820,7 +820,12 @@ function test_rrule_performance(
 
         # Test allocations in primal.
         f(x...)
-        @test (@allocations f(x...)) == 0
+
+        @static if VERSION >= v"1.12-"
+            @test Base.allocations(f, x...) == 0
+        else
+            @test (@allocations f(x...)) == 0
+        end
 
         # Test allocations in round-trip.
         f_f̄_fwds = to_fwds(f_f̄)

src/utils.jl

@@ -377,15 +377,15 @@ function misty_closure(
 end
 
 @unstable begin
-@static if VERSION > v"1.12-"
-    compute_ir_rettype(ir) = CC.compute_ir_rettype(ir)
-    compute_oc_signature(ir, nargs, isva) = CC.compute_oc_signature(ir, nargs, isva)
-else
-    compute_ir_rettype(ir) = Base.Experimental.compute_ir_rettype(ir)
-    compute_oc_signature(ir, nargs, isva) = Base.Experimental.compute_oc_signature(
-        ir, nargs, isva
-    )
-end
+    @static if VERSION > v"1.12-"
+        compute_ir_rettype(ir) = CC.compute_ir_rettype(ir)
+        compute_oc_signature(ir, nargs, isva) = CC.compute_oc_signature(ir, nargs, isva)
+    else
+        compute_ir_rettype(ir) = Base.Experimental.compute_ir_rettype(ir)
+        compute_oc_signature(ir, nargs, isva) = Base.Experimental.compute_oc_signature(
+            ir, nargs, isva
+        )
+    end
 end
 
 """