Test that simplification now works after AD thanks to only local sharing

simplified/HordeAd/Core/AstSimplify.hs

@@ -21,7 +21,7 @@ module HordeAd.Core.AstSimplify
   , astConstant, astSum, astScatter, astFromList, astFromVector, astKonst
   , astAppend, astSlice, astReverse, astFromDynamic
   , astIntCond
-  , ShowAstSimplify, simplifyAst
+  , ShowAstSimplify, simplifyAst, simplifyAstDomains
   , substituteAst, substituteAstInt, substituteAstBool
   , resetVarCounter
   ) where

test/simplified/TestAdaptorSimplified.hs

@@ -392,6 +392,10 @@ testReluPP2 = do
     @?= length "\\s0 dt x3 -> dlet (tkonst 5 (s0 ! [0])) (\\x6 -> dlet (x3 * x6) (\\x7 -> dlet (tgather [5] (tconst (fromList [2] [0.0,1.0])) (\\[i5] -> [ifB (tlet (s0 ! [0]) (\\x12 -> tlet (x3 ! [i5]) (\\x13 -> x13 * x12)) <=* tconst 0.0) 0 1])) (\\x8 -> dlet (x8 * dt) (\\x9 -> dlet (tscatter [1] (tfromList [tsum (x3 * x9)]) (\\[i10] -> [0])) (\\x11 -> dmkDomains (fromList [dfromR (tfromList [tconst 0.0 + x11 ! [0]]), dfromR (x6 * x9)]))))))"
   length ("\\" ++ varsPP ++ " -> " ++ printAstSimple renames vAst)
     @?= length "\\s0 x3 -> tlet (tkonst 5 (s0 ! [0])) (\\x6 -> tlet (x3 * x6) (\\x7 -> tlet (tgather [5] (tconst (fromList [2] [0.0,1.0])) (\\[i5] -> [ifB (tlet (s0 ! [0]) (\\x12 -> tlet (x3 ! [i5]) (\\x13 -> x13 * x12)) <=* tconst 0.0) 0 1])) (\\x8 -> x8 * x7)))"
+  length ("\\" ++ varsPPD
+       ++ " -> " ++ printAstDomainsPretty renames
+                                          (simplifyAstDomains letGradientAst))
+    @?= length "\\s0 dt x3 -> let x6 = tkonst 5 (s0 ! [0]) in let x7 = x3 * x6 in let x8 = tconstant (tgather [5] (tconst (fromList [2] [0.0,1.0])) (\\[i5] -> [ifB ((let x12 = s0 ! [0] in let x13 = x3 ! [i5] in x13 * x12) <=* tconst 0.0) 0 1])) in let x9 = x8 * dt in let x11 = tscatter [1] (tkonst 1 (tsum (x3 * x9))) (\\[i10] -> [0]) in (tkonst 1 (tconst 0.0 + x11 ! [0]), x6 * x9)"
 
 reluMax :: forall n r. (ADReady r, KnownNat n)
         => TensorOf n r -> TensorOf n r