Skip to content

Commit eaa1051

Browse files
author
marcrasi
authored
[AutoDiff] make differentiable array ops inlinable (#31704)
1 parent d613cd9 commit eaa1051

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

stdlib/public/Differentiation/ArrayDifferentiation.swift

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -262,6 +262,7 @@ extension Array where Element: Differentiable {
262262
//===----------------------------------------------------------------------===//
263263

264264
extension Array where Element: Differentiable {
265+
@inlinable
265266
@differentiable(wrt: (self, initialResult))
266267
public func differentiableReduce<Result: Differentiable>(
267268
_ initialResult: Result,
@@ -270,7 +271,7 @@ extension Array where Element: Differentiable {
270271
reduce(initialResult, nextPartialResult)
271272
}
272273

273-
@usableFromInline
274+
@inlinable
274275
@derivative(of: differentiableReduce)
275276
internal func _vjpDifferentiableReduce<Result: Differentiable>(
276277
_ initialResult: Result,
@@ -310,14 +311,15 @@ extension Array where Element: Differentiable {
310311
}
311312

312313
extension Array where Element: Differentiable {
314+
@inlinable
313315
@differentiable(wrt: self)
314316
public func differentiableMap<Result: Differentiable>(
315317
_ body: @differentiable (Element) -> Result
316318
) -> [Result] {
317319
map(body)
318320
}
319321

320-
@usableFromInline
322+
@inlinable
321323
@derivative(of: differentiableMap)
322324
internal func _vjpDifferentiableMap<Result: Differentiable>(
323325
_ body: @differentiable (Element) -> Result

0 commit comments

Comments
 (0)