Gardening.

- Remove unused SWIFT_ENABLE_TENSORFLOW known protocols.
- Readd AutoDiff test, previously accidentally removed.

Author: dan-zheng

include/swift/AST/KnownProtocols.def

@@ -78,8 +78,6 @@ PROTOCOL(Encodable)
 PROTOCOL(Decodable)
 // SWIFT_ENABLE_TENSORFLOW
 PROTOCOL(AdditiveArithmetic)
-PROTOCOL(Numeric)
-PROTOCOL(FloatingPoint)
 PROTOCOL(KeyPathIterable)
 PROTOCOL(TensorArrayProtocol)
 PROTOCOL(TensorGroup)

lib/IRGen/GenMeta.cpp

@@ -4189,7 +4189,6 @@ SpecialProtocol irgen::getSpecialProtocolID(ProtocolDecl *P) {
   case KnownProtocolKind::ExpressibleByColorLiteral:
   case KnownProtocolKind::ExpressibleByImageLiteral:
   case KnownProtocolKind::ExpressibleByFileReferenceLiteral:
-  // SWIFT_ENABLE_TENSORFLOW
   case KnownProtocolKind::ExpressibleByBuiltinBooleanLiteral:
   case KnownProtocolKind::ExpressibleByBuiltinExtendedGraphemeClusterLiteral:
   case KnownProtocolKind::ExpressibleByBuiltinFloatLiteral:
@@ -4206,9 +4205,7 @@ SpecialProtocol irgen::getSpecialProtocolID(ProtocolDecl *P) {
   case KnownProtocolKind::Decodable:
   case KnownProtocolKind::StringInterpolationProtocol:
   // SWIFT_ENABLE_TENSORFLOW
-  case KnownProtocolKind::FloatingPoint:
   case KnownProtocolKind::AdditiveArithmetic:
-  case KnownProtocolKind::Numeric:
   case KnownProtocolKind::KeyPathIterable:
   case KnownProtocolKind::TensorArrayProtocol:
   case KnownProtocolKind::TensorGroup:

lib/SILOptimizer/Mandatory/Differentiation.cpp

@@ -865,18 +865,12 @@ class ADContext {
   /// Saved for deletion during cleanup.
   SmallVector<SILValue, 32> generatedAssociatedFunctionReferences;
 
+  /// The AdditiveArithmetic protocol in the standard library.
+  ProtocolDecl *additiveArithmeticProtocol =
+      astCtx.getProtocol(KnownProtocolKind::AdditiveArithmetic);
   /// The VectorProtocol protocol in the standard library.
   ProtocolDecl *vectorProtocolProtocol =
       astCtx.getProtocol(KnownProtocolKind::VectorProtocol);
-  /// The Numeric protocol in the standard library.
-  ProtocolDecl *numericProtocol =
-      astCtx.getProtocol(KnownProtocolKind::Numeric);
-  /// The AdditiveArithmetic protocol in the standard library.
-  ProtocolDecl *additiveArithmeticProtocol =
-    astCtx.getProtocol(KnownProtocolKind::AdditiveArithmetic);
-  /// The FloatingPoint protocol in the stanard library.
-  ProtocolDecl *floatingPointProtocol =
-      astCtx.getProtocol(KnownProtocolKind::FloatingPoint);
 
   /// `AdditiveArithmetic.+` declaration.
   mutable FuncDecl *cachedPlusFn = nullptr;
@@ -926,20 +920,12 @@ class ADContext {
     return generatedAssociatedFunctionReferences;
   }
 
-  ProtocolDecl *getVectorProtocolProtocol() const {
-    return vectorProtocolProtocol;
-  }
-
-  ProtocolDecl *getNumericProtocol() const {
-    return numericProtocol;
-  }
-
   ProtocolDecl *getAdditiveArithmeticProtocol() const {
     return additiveArithmeticProtocol;
   }
 
-  ProtocolDecl *getFloatingPointProtocol() const {
-    return floatingPointProtocol;
+  ProtocolDecl *getVectorProtocolProtocol() const {
+    return vectorProtocolProtocol;
   }
 
   FuncDecl *getPlusDecl() const {

test/AutoDiff/separate_tangent_type.swift

@@ -0,0 +1,49 @@
+// RUN: %target-run-simple-swift
+// REQUIRES: executable_test
+
+import StdlibUnittest
+#if os(macOS)
+import Darwin.C
+#else
+import Glibc
+#endif
+
+var SeparateTangentTypeTests = TestSuite("SeparateTangentType")
+
+struct DifferentiableSubset : Differentiable {
+  @differentiable(wrt: self)
+  var w: Float
+  @differentiable(wrt: self)
+  var b: Float
+  @noDerivative var flag: Bool
+
+  struct TangentVector : Differentiable, VectorProtocol {
+    typealias TangentVector = DifferentiableSubset.TangentVector
+    var w: Float
+    var b: Float
+  }
+  mutating func move(along v: TangentVector) {
+    w.move(along: v.w)
+    b.move(along: v.b)
+  }
+}
+
+SeparateTangentTypeTests.test("Trivial") {
+  let x = DifferentiableSubset(w: 0, b: 1, flag: false)
+  let pb = pullback(at: x) { x in x }
+  expectEqual(pb(DifferentiableSubset.TangentVector.zero), DifferentiableSubset.TangentVector.zero)
+}
+
+SeparateTangentTypeTests.test("Initialization") {
+  let x = DifferentiableSubset(w: 0, b: 1, flag: false)
+  let pb = pullback(at: x) { x in DifferentiableSubset(w: 1, b: 2, flag: true) }
+  expectEqual(pb(DifferentiableSubset.TangentVector.zero), DifferentiableSubset.TangentVector.zero)
+}
+
+SeparateTangentTypeTests.test("SomeArithmetics") {
+  let x = DifferentiableSubset(w: 0, b: 1, flag: false)
+  let pb = pullback(at: x) { x in DifferentiableSubset(w: x.w * x.w, b: x.b * x.b, flag: true) }
+  expectEqual(pb(DifferentiableSubset.TangentVector.zero), DifferentiableSubset.TangentVector.zero)
+}
+
+runAllTests()