From ec1248b82629da473b4c1603830c518b9ae4befd Mon Sep 17 00:00:00 2001
From: Bruno Ploumhans <13494793+Technici4n@users.noreply.github.com>
Date: Thu, 6 Nov 2025 21:54:44 +0100
Subject: [PATCH] Try to use count_allocs instead of @allocations

---
 src/test_utils.jl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/test_utils.jl b/src/test_utils.jl
index 4585eb428..9a350de41 100644
--- a/src/test_utils.jl
+++ b/src/test_utils.jl
@@ -820,7 +820,7 @@ function test_rrule_performance(
 
         # Test allocations in primal.
         f(x...)
-        @test (@allocations f(x...)) == 0
+        @test count_allocs(f, x...) == 0
 
         # Test allocations in round-trip.
         f_f̄_fwds = to_fwds(f_f̄)