Skip to content

Commit 60d52ba

Browse files
committed
fix: ref negate
1 parent 87aa9a8 commit 60d52ba

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

lib/LuxLib/test/common_ops/dense_tests.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -138,7 +138,7 @@ end
138138
end
139139

140140
@testitem "Enzyme.Forward patch: dense" tags = [:common] setup = [SharedTestSetup] skip =
141-
:(using LuxTestUtils; !LuxTestUtils.ENZYME_TESTING_ENABLED) begin
141+
:(using LuxTestUtils; !LuxTestUtils.ENZYME_TESTING_ENABLED[]) begin
142142
using LuxLib, Random, ForwardDiff, Enzyme
143143

144144
x = rand(Float32, 2, 2)
@@ -149,7 +149,7 @@ end
149149
end
150150

151151
@testitem "Enzyme rules for fused dense" tags = [:common] setup = [SharedTestSetup] skip =
152-
:(using LuxTestUtils; !LuxTestUtils.ENZYME_TESTING_ENABLED) begin
152+
:(using LuxTestUtils; !LuxTestUtils.ENZYME_TESTING_ENABLED[]) begin
153153
using LuxLib, NNlib, Zygote, Enzyme
154154

155155
# These are mostly for testing the CUDA rules since we don't enable the CUDA tests

test/helpers/training_tests.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ end
4040

4141
for ad in (AutoZygote(), AutoTracker(), AutoReverseDiff(), AutoEnzyme())
4242
ongpu && (ad isa AutoReverseDiff || ad isa AutoEnzyme) && continue
43-
!LuxTestUtils.ENZYME_TESTING_ENABLED && ad isa AutoEnzyme && continue
43+
!LuxTestUtils.ENZYME_TESTING_ENABLED[] && ad isa AutoEnzyme && continue
4444

4545
grads, _, _, _ = Training.compute_gradients(ad, _loss_function, x, tstate)
4646
tstate_ = Training.apply_gradients(tstate, grads)
@@ -192,7 +192,7 @@ end
192192
end
193193

194194
@testitem "Training API Enzyme Runtime Mode" setup = [SharedTestSetup] tags = [:misc] skip =
195-
:(using LuxTestUtils; !LuxTestUtils.ENZYME_TESTING_ENABLED) begin
195+
:(using LuxTestUtils; !LuxTestUtils.ENZYME_TESTING_ENABLED[]) begin
196196
using Lux, Random, Enzyme, Optimisers
197197

198198
function makemodel(n)
@@ -241,7 +241,7 @@ end
241241

242242
@testitem "Enzyme: Invalidate Cache on State Update" setup = [SharedTestSetup] tags = [
243243
:misc
244-
] skip = :(using LuxTestUtils; !LuxTestUtils.ENZYME_TESTING_ENABLED) begin
244+
] skip = :(using LuxTestUtils; !LuxTestUtils.ENZYME_TESTING_ENABLED[]) begin
245245
using ADTypes, Optimisers
246246

247247
mse = MSELoss()

0 commit comments

Comments
 (0)