Skip to content

Commit a2c214c

Browse files
committed
Try to pacify Linux
1 parent 869bc1a commit a2c214c

File tree

1 file changed

+2
-39
lines changed

1 file changed

+2
-39
lines changed

test/AutoDiff/SILOptimizer/BuildingSimulation.swift

Lines changed: 2 additions & 39 deletions
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,12 @@
11
// RUN: %target-swift-frontend -emit-sil -verify -O %s | %FileCheck %s
22
// REQUIRES: swift_in_compiler
3-
// UNSUPPORTED: OS=windows-msvc
43

54
import _Differentiation
6-
#if canImport(Glibc)
7-
import Glibc
8-
#else
9-
import Foundation
10-
#endif
115

126
// Simulation parameters
137
let trials = 100
148
let timesteps = 20
159
let dTime: Float = 0.1
16-
let printGradToCompare = false
1710

1811
// Definitions
1912
let π = Float.pi
@@ -219,47 +212,17 @@ func dontLetTheCompilerOptimizeThisAway<T>(_ x: T) {
219212
blackHole = x
220213
}
221214

222-
func measure<T>(_ block: () throws -> T) throws -> (time: Double, result: T) {
223-
let t0 = DispatchTime.now()
224-
let result = try block()
225-
let t1 = DispatchTime.now()
226-
let elapsed = Double(t1.uptimeNanoseconds - t0.uptimeNanoseconds) / 1E9
227-
return (elapsed, result)
228-
}
229-
230215
@differentiable(reverse)
231216
func fullPipe(simParams: SimParams) -> Float {
232217
let pred = simulate(simParams: simParams)
233218
let loss = lossCalc(pred: pred, gt: 27.344767)
234219
return loss
235220
}
236221

237-
var totalPureForwardTime: Double = 0
238-
var totalGradientTime: Double = 0
239-
240222
for _ in 0 ..< trials {
241-
let (forwardOnly, _) = try measure {
242-
return fullPipe(simParams: simParams)
243-
}
223+
let forwardOnly = fullPipe(simParams: simParams)
244224
dontLetTheCompilerOptimizeThisAway(forwardOnly)
245225

246-
let (gradientTime, grad) = try measure {
247-
return gradient(at: simParams, of: fullPipe)
248-
}
226+
let grad = gradient(at: simParams, of: fullPipe)
249227
dontLetTheCompilerOptimizeThisAway(grad)
250-
251-
if printGradToCompare {
252-
print(grad)
253-
}
254-
255-
totalPureForwardTime += forwardOnly
256-
totalGradientTime += gradientTime
257228
}
258-
259-
let averagePureForward = totalPureForwardTime / Double(trials)
260-
let averageGradient = totalGradientTime / Double(trials)
261-
262-
print("trials: \(trials)")
263-
print("timesteps: \(timesteps)")
264-
print("average forward only time: \(averagePureForward) seconds")
265-
print("average forward and back (gradient) time: \(averageGradient) seconds")

0 commit comments

Comments
 (0)