TensorFlow: annotate functions using _forEachFieldWithKeyPath

Because the standard toolchain support requires the newest runtime in
order to have access to `_forEachFieldWithKeyPath` and the runtime is
bundled into the OS on macOS, we need to annotate the functions with
availability.

Author: compnerd

Sources/TensorFlow/Core/ElementaryFunctions.swift

@@ -13,9 +13,12 @@
 // limitations under the License.
 
 #if TENSORFLOW_USE_STANDARD_TOOLCHAIN
+
 import Numerics
 @_spi(Reflection) import Swift
+
 extension ElementaryFunctions {
+  @available(macOS 9999, *)
   internal static func visitChildren(
     _ body: (PartialKeyPath<Self>, ElementaryFunctionsVisit.Type) -> Void
   ) {
@@ -164,4 +167,5 @@ extension ElementaryFunctions {
   public static func root(_ x: Self, _ n: Int) -> Self { .init(mapped: Functor_root(n: n), x) }
   public static func pow(_ x: Self, _ y: Self) -> Self { .init(mapped: Functor_pow2(), x, y) }
 }
+
 #endif

Sources/TensorFlow/Core/EuclideanDifferentiable.swift

@@ -15,8 +15,10 @@
 import _Differentiation
 
 #if TENSORFLOW_USE_STANDARD_TOOLCHAIN
+
 @_spi(Reflection) import Swift
 
+@available(macOS 9999, *)
 func listFields<Root>(of type: Root.Type) -> [(String, PartialKeyPath<Root>)] {
   var out = [(String, PartialKeyPath<Root>)]()
   _forEachFieldWithKeyPath(of: type, options: .ignoreUnknown) { name, kp in
@@ -27,8 +29,8 @@ func listFields<Root>(of type: Root.Type) -> [(String, PartialKeyPath<Root>)] {
 }
 
 extension Differentiable {
-  static var differentiableFields: [(String, PartialKeyPath<Self>, PartialKeyPath<TangentVector>)]
-  {
+  @available(macOS 9999, *)
+  static var differentiableFields: [(String, PartialKeyPath<Self>, PartialKeyPath<TangentVector>)] {
     let tangentFields = listFields(of: TangentVector.self)
     var i = 0
     var out = [(String, PartialKeyPath<Self>, PartialKeyPath<TangentVector>)]()

Sources/TensorFlow/Core/KeyPathIterable.swift

@@ -21,6 +21,7 @@
 import _Differentiation
 
 #if TENSORFLOW_USE_STANDARD_TOOLCHAIN
+
 @_spi(Reflection) import Swift
 
 /// An implementation detail of `KeyPathIterable`; do not use this protocol
@@ -41,6 +42,7 @@ public protocol KeyPathIterable: _KeyPathIterableBase {
 }
 
 public extension KeyPathIterable {
+  @available(macOS 9999, *)
   var allKeyPaths: [PartialKeyPath<Self>] {
     var out = [PartialKeyPath<Self>]()
     _forEachFieldWithKeyPath(of: Self.self, options: .ignoreUnknown) { name, kp in
@@ -171,4 +173,5 @@ extension Optional.TangentVector: KeyPathIterable {
     return []
   }
 }
+
 #endif

Sources/TensorFlow/Core/PointwiseMultiplicative.swift

@@ -15,6 +15,7 @@
 import _Differentiation
 
 #if TENSORFLOW_USE_STANDARD_TOOLCHAIN
+
 @_spi(Reflection) import Swift
 
 infix operator .*: MultiplicationPrecedence
@@ -111,6 +112,7 @@ extension PointwiseMultiplicative {
 }
 
 extension PointwiseMultiplicative {
+  @available(macOS 9999, *)
   internal static func visitChildren(
     _ body: (PartialKeyPath<Self>, _PointwiseMultiplicative.Type) -> Void
   ) {
@@ -134,4 +136,5 @@ extension PointwiseMultiplicative {
 extension Array.DifferentiableView: _PointwiseMultiplicative
 where Element: Differentiable & PointwiseMultiplicative {}
 extension Tensor: _PointwiseMultiplicative where Scalar: Numeric {}
+
 #endif

Sources/TensorFlow/Core/TensorGroup.swift

@@ -340,8 +340,10 @@ extension Array: TensorArrayProtocol where Element: TensorGroup {
 }
 
 #if TENSORFLOW_USE_STANDARD_TOOLCHAIN
+
 @_spi(Reflection) import Swift
 
+@available(macOS 9999, *)
 func reflectionInit<T>(type: T.Type, body: (inout T, PartialKeyPath<T>) -> Void) -> T {
   let x = UnsafeMutablePointer<T>.allocate(capacity: 1)
   defer { x.deallocate() }
@@ -355,6 +357,7 @@ func reflectionInit<T>(type: T.Type, body: (inout T, PartialKeyPath<T>) -> Void)
 }
 
 extension TensorGroup {
+  @available(macOS 9999, *)
   public static var _typeList: [TensorDataType] {
     var out = [TensorDataType]()
     if !(_forEachFieldWithKeyPath(of: Self.self) { name, kp in
@@ -366,6 +369,7 @@ extension TensorGroup {
     }
     return out
   }
+
   public static func initialize<Root>(
     _ base: inout Root, _ kp: PartialKeyPath<Root>,
     _owning tensorHandles: UnsafePointer<CTensorHandle>?
@@ -377,6 +381,7 @@ extension TensorGroup {
       v.initialize(to: .init(_owning: tensorHandles))
     }
   }
+
   public init(_owning tensorHandles: UnsafePointer<CTensorHandle>?) {
     var i = 0
     self = reflectionInit(type: Self.self) { base, kp in
@@ -387,6 +392,8 @@ extension TensorGroup {
       i += Int(valueType._tensorHandleCount)
     }
   }
+
+  @available(macOS 9999, *)
   public func _unpackTensorHandles(into address: UnsafeMutablePointer<CTensorHandle>?) {
     var i = 0
     if !_forEachFieldWithKeyPath(of: Self.self, body: { name, kp in
@@ -399,4 +406,5 @@ extension TensorGroup {
     }
   }
 }
+
 #endif

Sources/TensorFlow/Core/VectorProtocol.swift

@@ -15,6 +15,7 @@
 import _Differentiation
 
 #if TENSORFLOW_USE_STANDARD_TOOLCHAIN
+
 @_spi(Reflection) import Swift
 
 /// Implementation detail for reflection.
@@ -35,6 +36,7 @@ public protocol _VectorProtocol {
 }
 
 extension VectorProtocol {
+  @available(macOS 9999, *)
   internal static func visitChildren(
     _ body: (PartialKeyPath<Self>, _VectorProtocol.Type) -> Void
   ) {
@@ -127,4 +129,5 @@ extension VectorProtocol {
 extension Tensor: _VectorProtocol where Scalar: TensorFlowFloatingPoint {}
 extension Array.DifferentiableView: _VectorProtocol
 where Element: Differentiable & VectorProtocol {}
+
 #endif