add option to print gradient for verification

Author: porterchild

Benchmarks/BuildingSimulation/PyTorch/PyTorchSimulator.py

@@ -177,6 +177,7 @@ def fullPipe(simParams):
 timesteps = 20
 trials = 30
 warmup = 3
+printGradToCompare = False
 
 for i in range(trials):
 
@@ -190,6 +191,9 @@ def getGradient(simParams):
 
 
     gradientTime, gradient = measure(getGradient, simParams)
+
+    if printGradToCompare:
+        print(gradient)
 
     if i >= warmup:
         totalForwardTime += forwardTime

Benchmarks/BuildingSimulation/Swift/main.swift

@@ -220,6 +220,7 @@ var trials = 30
 var timesteps = 20
 var totalPureForwardTime: Double = 0
 var totalGradientTime: Double = 0
+let printGradToCompare = false
 
 for _ in 0 ..< trials {
     let (forwardOnly, _) = try measure {
@@ -232,6 +233,8 @@ for _ in 0 ..< trials {
     }
     dontLetTheCompilerOptimizeThisAway(grad)
 
+    if printGradToCompare {
+        print(grad)
     }
 
     totalPureForwardTime += forwardOnly

Benchmarks/BuildingSimulation/TensorFlow/TensorFlowSimulator.py

@@ -227,6 +227,7 @@ def fullPipe(simParams):
 timesteps = 20
 trials = 30
 warmup = 3
+printGradToCompare = False
 
 for i in range(trials + warmup):
 
@@ -243,6 +244,9 @@ def getGradient(simParams):
 
     gradientTime, gradient = measure(getGradient, simParams)
 
+    if printGradToCompare:
+        print(gradient)
+
     if i >= warmup:
         totalForwardTime += forwardTime
         totalGradientTime += gradientTime