Add requested test.

Fix `ref_element_addr` adjoint buffer type remapping.
Garden tests.

Author: dan-zheng

lib/SILOptimizer/Utils/Differentiation/PullbackEmitter.cpp

@@ -376,7 +376,8 @@ SILValue PullbackEmitter::getAdjointProjection(SILBasicBlock *origBB,
     auto *tanField = cast<VarDecl>(tanFieldLookup.front());
     // Create a local allocation for the element adjoint buffer.
     auto eltTanType = tanField->getValueInterfaceType()->getCanonicalType();
-    auto eltTanSILType = SILType::getPrimitiveAddressType(eltTanType);
+    auto eltTanSILType =
+        remapType(SILType::getPrimitiveAddressType(eltTanType));
     auto *eltAdjBuffer = createFunctionLocalAllocation(eltTanSILType, loc);
     builder.emitScopedBorrowOperation(
         loc, adjClass, [&](SILValue borrowedAdjClass) {
@@ -1090,7 +1091,7 @@ PullbackEmitter::getArrayAdjointElementBuffer(SILValue arrayAdjoint,
   auto arrayTanType = cast<StructType>(arrayAdjoint->getType().getASTType());
   auto arrayType = arrayTanType->getParent()->castTo<BoundGenericStructType>();
   auto eltTanType = arrayType->getGenericArgs().front()->getCanonicalType();
-  auto eltTanSILType = SILType::getPrimitiveAddressType(eltTanType);
+  auto eltTanSILType = remapType(SILType::getPrimitiveAddressType(eltTanType));
   // Get `function_ref` and generic signature of
   // `Array.TangentVector.subscript.getter`.
   auto *arrayTanStructDecl = arrayTanType->getStructOrBoundGenericStruct();
@@ -1602,12 +1603,11 @@ void PullbackEmitter::visitLoadOperation(SingleValueInstruction *inst) {
 void PullbackEmitter::visitStoreOperation(SILBasicBlock *bb, SILLocation loc,
                                           SILValue origSrc, SILValue origDest) {
   auto &adjBuf = getAdjointBuffer(bb, origDest);
-  auto bufType = remapType(adjBuf->getType());
   auto adjVal =
       builder.emitLoadValueOperation(loc, adjBuf, LoadOwnershipQualifier::Take);
   recordTemporary(adjVal);
   addAdjointValue(bb, origSrc, makeConcreteAdjointValue(adjVal), loc);
-  emitZeroIndirect(bufType.getASTType(), adjBuf, loc);
+  emitZeroIndirect(adjBuf->getType().getASTType(), adjBuf, loc);
 }
 
 void PullbackEmitter::visitStoreInst(StoreInst *si) {

test/AutoDiff/downstream/class_differentiation.swift

@@ -16,10 +16,16 @@ ClassTests.test("TrivialMember") {
     @noDerivative
     final var noDerivative: Float = 1
 
+    @differentiable
     init(_ float: Float) {
       self.float = float
     }
 
+    @differentiable
+    convenience init(convenience x: Float) {
+      self.init(x)
+    }
+
     @differentiable
     func method(_ x: Float) -> Float {
       x * float
@@ -44,6 +50,7 @@ ClassTests.test("TrivialMember") {
   }
   // Test class initializer differentiation.
   expectEqual(10, pullback(at: 3, in: { C($0) })(.init(float: 10)))
+  expectEqual(10, pullback(at: 3, in: { C(convenience: $0) })(.init(float: 10)))
   // Test class method differentiation.
   expectEqual((.init(float: 3), 10), gradient(at: C(10), 3, in: { c, x in c.method(x) }))
   expectEqual(.init(float: 0), gradient(at: C(10), in: { c in c.testNoDerivative() }))
@@ -56,6 +63,7 @@ ClassTests.test("NontrivialMember") {
     @differentiable
     var float: Tracked<Float>
 
+    @differentiable
     init(_ float: Tracked<Float>) {
       self.float = float
     }
@@ -84,6 +92,26 @@ ClassTests.test("NontrivialMember") {
               gradient(at: C(10), C(20), in: { c1, c2 in C.controlFlow(c1, c2, true) }))
 }
 
+ClassTests.test("GenericNontrivialMember") {
+  final class C<T: Differentiable>: Differentiable where T == T.TangentVector {
+    @differentiable
+    var x: Tracked<T>
+
+    @differentiable
+    init(_ x: T) {
+      self.x = Tracked(x)
+    }
+
+    @differentiable
+    convenience init(convenience x: T) {
+      self.init(x)
+    }
+  }
+  // Test class initializer differentiation.
+  expectEqual(10, pullback(at: 3, in: { C<Float>($0) })(.init(x: 10)))
+  expectEqual(10, pullback(at: 3, in: { C<Float>(convenience: $0) })(.init(x: 10)))
+}
+
 // TF-1149: Test class with loadable type but address-only `TangentVector` type.
 // TODO(TF-1149): Uncomment when supported.
 /*