Merge pull request #36124 from rxwei/remove-withderivative

Author: swift-ci

stdlib/private/DifferentiationUnittest/DifferentiationUnittest.swift.gyb

@@ -34,6 +34,38 @@ public func withLeakChecking(
     file: file, line: line)
 }
 
+@inlinable
+@_semantics("autodiff.nonvarying")
+public func withoutDerivative<T, R>(at x: T, in body: (T) -> R) -> R
+{
+  body(x)
+}
+
+public extension Differentiable {
+  /// Applies the given closure to the derivative of `self`.
+  ///
+  /// Returns `self` like an identity function. When the return value is used in
+  /// a context where it is differentiated with respect to, applies the given
+  /// closure to the derivative of the return value.
+  @inlinable
+  @differentiable(reverse, wrt: self)
+  func withDerivative(_ body: @escaping (inout TangentVector) -> Void) -> Self {
+    return self
+  }
+
+  @inlinable
+  @derivative(of: withDerivative)
+  internal func _vjpWithDerivative(
+    _ body: @escaping (inout TangentVector) -> Void
+  ) -> (value: Self, pullback: (TangentVector) -> TangentVector) {
+    return (self, { grad in
+      var grad = grad
+      body(&grad)
+      return grad
+    })
+  }
+}
+
 public extension TestSuite {
   /// Execute test function and check expected leak count.
   func testWithLeakChecking(

stdlib/public/Differentiation/DifferentiationUtilities.swift

@@ -31,42 +31,6 @@ public func withoutDerivative<T>(at x: T) -> T {
   x
 }
 
-/// Applies the given closure `body` to `x`. When used in a context where `x` is
-/// being differentiated with respect to, this function will not produce any
-/// derivative at `x`.
-// FIXME: Support throws-rethrows.
-@inlinable
-@inline(__always)
-@_semantics("autodiff.nonvarying")
-public func withoutDerivative<T, R>(at x: T, in body: (T) -> R) -> R {
-  body(x)
-}
-
-public extension Differentiable {
-  /// Applies the given closure to the derivative of `self`.
-  ///
-  /// Returns `self` like an identity function. When the return value is used in
-  /// a context where it is differentiated with respect to, applies the given
-  /// closure to the derivative of the return value.
-  @inlinable
-  @differentiable(reverse, wrt: self)
-  func withDerivative(_ body: @escaping (inout TangentVector) -> Void) -> Self {
-    return self
-  }
-
-  @inlinable
-  @derivative(of: withDerivative)
-  internal func _vjpWithDerivative(
-    _ body: @escaping (inout TangentVector) -> Void
-  ) -> (value: Self, pullback: (TangentVector) -> TangentVector) {
-    return (self, { grad in
-      var grad = grad
-      body(&grad)
-      return grad
-    })
-  }
-}
-
 //===----------------------------------------------------------------------===//
 // Diagnostics
 //===----------------------------------------------------------------------===//