mirror of
https://github.com/apple/swift.git
synced 2026-10-26 20:15:53 +01:00
Anton Korobeynikov
ca483d0fa0
This effectively allows us to register single derivative for a multiple original functions: protocol requirement and (possibly) default implementations from extensions. As a result, we might have default derivatives registered for protocol requirements and non-differentiable requirements conditionally differentiable (in an extension). Roll back test changes from #84800 and re-enable `SIMD` and `Tracked<T>` differentiability by providing default derivatives for `AdditiveArithmetic` and `Numeric`. Fixes #55611 Fixes #84840 Fixes #87705 Fixes #87693
96 lines
2.8 KiB
Swift
96 lines
2.8 KiB
Swift
//===--- AdditiveArithmeticDifferentiation.swift--- ----------*- swift -*-===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 2026 Apple Inc. and the Swift project authors
|
|
// Licensed under Apache License v2.0 with Runtime Library Exception
|
|
//
|
|
// See https://swift.org/LICENSE.txt for license information
|
|
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
import Swift
|
|
|
|
extension AdditiveArithmetic where Self: Differentiable {
|
|
@derivative(of: +)
|
|
@inlinable
|
|
@_transparent
|
|
static func _vjpPlus(_ x: Self) ->
|
|
(value: Self, pullback: (TangentVector) -> (TangentVector)) {
|
|
return (x, { v in v })
|
|
}
|
|
|
|
@inlinable
|
|
@_transparent
|
|
@derivative(of: +)
|
|
static func _jvpPlus(x: Self) ->
|
|
(value: Self, differential: (TangentVector) -> TangentVector) {
|
|
return (x, { dx in dx })
|
|
}
|
|
|
|
@derivative(of: +)
|
|
@inlinable
|
|
@_transparent
|
|
static func _vjpAdd(_ lhs: Self, _ rhs: Self) ->
|
|
(value: Self, pullback: (TangentVector) -> (TangentVector, TangentVector)) {
|
|
return (lhs + rhs, { v in (v, v) })
|
|
}
|
|
|
|
@inlinable
|
|
@_transparent
|
|
@derivative(of: +)
|
|
static func _jvpAdd(lhs: Self, rhs: Self) ->
|
|
(value: Self, differential: (TangentVector, TangentVector) -> TangentVector) {
|
|
return (lhs + rhs, { (dlhs, drhs) in dlhs + drhs })
|
|
}
|
|
|
|
@export(implementation)
|
|
@derivative(of: +=)
|
|
static func _vjpAddAssign(_ lhs: inout Self, _ rhs: Self) ->
|
|
(value: Void, pullback: (inout TangentVector) -> TangentVector) {
|
|
lhs += rhs
|
|
return ((), { v in v })
|
|
}
|
|
|
|
@export(implementation)
|
|
@derivative(of: +=)
|
|
static func _jvpAddAssign(_ lhs: inout Self, _ rhs: Self) ->
|
|
(value: Void, differential: (inout TangentVector, TangentVector) -> Void) {
|
|
lhs += rhs
|
|
return ((), { $0 += $1 })
|
|
}
|
|
|
|
@inlinable
|
|
@_transparent
|
|
@derivative(of: -)
|
|
static func _vjpSubtract(lhs: Self, rhs: Self) ->
|
|
(value: Self, pullback: (TangentVector) -> (TangentVector, TangentVector)) {
|
|
return (lhs - rhs, { v in (v, TangentVector.zero-v) })
|
|
}
|
|
|
|
@inlinable
|
|
@_transparent
|
|
@derivative(of: -)
|
|
static func _jvpSubtract(lhs: Self, rhs: Self) ->
|
|
(value: Self, differential: (TangentVector, TangentVector) -> TangentVector) {
|
|
return (lhs - rhs, { (dlhs, drhs) in dlhs - drhs })
|
|
}
|
|
|
|
@export(implementation)
|
|
@derivative(of: -=)
|
|
static func _vjpSubtractAssign(_ lhs: inout Self, _ rhs: Self) ->
|
|
(value: Void, pullback: (inout TangentVector) -> TangentVector) {
|
|
lhs -= rhs
|
|
return ((), { v in TangentVector.zero-v })
|
|
}
|
|
|
|
@export(implementation)
|
|
@derivative(of: -=)
|
|
static func _jvpSubtractAssign(_ lhs: inout Self, _ rhs: Self) ->
|
|
(value: Void, differential: (inout TangentVector, TangentVector) -> Void) {
|
|
lhs -= rhs
|
|
return ((), { $0 -= $1 })
|
|
}
|
|
}
|