From 27ca9845abac44a0ad7c776fdee1e89dd988f77b Mon Sep 17 00:00:00 2001
From: bendat78 <b.schemmer@medmems.de>
Date: Tue, 29 Aug 2023 22:47:52 +0200
Subject: [PATCH] gradient buffer not declared in mnist test

the gradientbuffer is commented out and then never used in training. Throws a compilation error when testing.
---
 test/mnist.jl | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/test/mnist.jl b/test/mnist.jl
index db2e4bc..302aa37 100644
--- a/test/mnist.jl
+++ b/test/mnist.jl
@@ -51,9 +51,11 @@ ENV["DATADEPS_ALWAYS_ACCEPT"] = "true"
 
   # initialize a gradient buffer matrix; number of columns places an upper bound
   # on the number of threads used.
-  # G = similar(p, length(p), min(Threads.nthreads(), (Sys.CPU_THREADS ÷ ((Sys.ARCH === :x86_64) + 1))));
+  G = similar(p, length(p), min(Threads.nthreads(), (Sys.CPU_THREADS ÷ ((Sys.ARCH === :x86_64) + 1))));
+  
   # train
   @time SimpleChains.train_batched!(
+    G,
     p,
     lenetloss,
     xtrain4,
@@ -62,6 +64,7 @@ ENV["DATADEPS_ALWAYS_ACCEPT"] = "true"
   )
   if VERSION >= v"1.10"
     @test_opt SimpleChains.train_batched!(
+      G,
       p,
       lenetloss,
       xtrain4,