Let the rest of the repo compile after the recent changes

Mikolaj · Feb 26, 2024 · 0a68927 · 0a68927
1 parent b7aaa9c
commit 0a68927
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/bench/common/BenchMnistTools.hs b/bench/common/BenchMnistTools.hs
@@ -53,7 +53,7 @@ mnistTrainBench1VTA extraPrefix chunkLength xs widthHidden widthHidden2
       f mnist adinputs =
         MnistFcnnRanked1.afcnnMnistLoss1
           widthHidden widthHidden2
-          mnist (parseHVector valsInit adinputs)
+          mnist (parseHVector (fromDValue valsInit) adinputs)
       chunk = take chunkLength xs
       grad c = fst $ sgd gamma f c hVectorInit
       name = extraPrefix
@@ -143,7 +143,7 @@ mnistTrainBench1VTO extraPrefix chunkLength xs widthHidden widthHidden2
         f = MnistFcnnRanked1.afcnnMnistLoss1TensorData @(AstRanked FullSpan)
               widthHidden widthHidden2
               (rconstant astGlyph, rconstant astLabel)
-        g hVector = f $ parseHVector valsInit hVector
+        g hVector = f $ parseHVector (fromValue valsInit) hVector
         (((varDtAgain, vars1Again), gradientRaw, primal, sh), _) =
           revProduceArtifact TensorToken False g envInit (voidFromHVector hVectorInit)
         gradient = simplifyAstHVector6 gradientRaw
@@ -221,7 +221,7 @@ mnistTrainBench2VTA extraPrefix chunkLength xs widthHidden widthHidden2
         -> ADVal ranked r 0
       f mnist adinputs =
         MnistFcnnRanked2.afcnnMnistLoss2
-          mnist (parseHVector valsInit adinputs)
+          mnist (parseHVector (fromDValue valsInit) adinputs)
       chunk = take chunkLength xs
       grad c = fst $ sgd gamma f c hVectorInit
       name = extraPrefix
@@ -316,7 +316,7 @@ mnistTrainBench2VTO extraPrefix chunkLength xs widthHidden widthHidden2
                   EM.empty
         f = MnistFcnnRanked2.afcnnMnistLoss2TensorData @(AstRanked FullSpan)
               (rconstant astGlyph, rconstant astLabel)
-        g hVector = f $ parseHVector valsInit hVector
+        g hVector = f $ parseHVector (fromValue valsInit) hVector
         (((varDtAgain, vars1Again), gradientRaw, primal, sh), _) =
           revProduceArtifact TensorToken False g envInit (voidFromHVector hVectorInit)
         gradient = simplifyAstHVector6 gradientRaw