averello/swift-mirror
1
0
Fork 0
mirror of https://github.com/apple/swift.git synced 2025-12-14 20:36:38 +01:00
Code Issues Packages Projects Releases Wiki Activity

Improve differentiableMap vjp performance (#83807)

Browse Source 
Improve differentiableMap vjp performance by reserving capacity for the values and pullbacks arrays.
This commit is contained in:
Jaap Wijnen
2025-08-20 02:08:14 +02:00
committed by GitHub
parent d837d6052c
commit a5ef5fe226
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
stdlib/public/Differentiation/ArrayDifferentiation.swift
View File

@@ -369,8 +369,11 @@ extension Array where Element: Differentiable {
value: [Result],
pullback: (Array<Result>.TangentVector) -> Array.TangentVector
) {
let count = self.count
var values: [Result] = []
var pullbacks: [(Result.TangentVector) -> Element.TangentVector] = []
values.reserveCapacity(count)
pullbacks.reserveCapacity(count)
for x in self {
let (y, pb) = valueWithPullback(at: x, of: body)
values.append(y)