Add another luxlib test

compintell · Oct 3, 2024 · 31daf96 · 31daf96
1 parent 2d54e51
commit 31daf96
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/test/ext/luxlib.jl b/test/ext/luxlib.jl
@@ -31,6 +31,21 @@
  LuxLib.Utils.True(),
  LuxLib.Utils.True(),
  ),
+ (
+ false, :none, false,
+ function(opmode, act, x, m, sigma2, gamma, beta)
+ LuxLib.Impl.batchnorm_affine_normalize_internal(
+ opmode, act, x, m, sigma2, gamma, beta, 1e-3
+ )
+ end,
+ LuxLib.LoopedArrayOp(),
+ Lux.relu,
+ randn(5, 4, 3),
+ randn(4),
+ rand(4) .+ 1.0,
+ nothing,
+ nothing,
+ ),
  ],
  vec(map(Iterators.product(
  [LuxLib.LoopedArrayOp(), LuxLib.GenericBroadcastOp()],