Fix various AD tests. (#24557)

- Add `[ossa]` tags.
- Add expected note for `differentiable_attr_type_checking.swift`.

Author: dan-zheng

test/AutoDiff/derived_differentiable_properties.swift

@@ -18,7 +18,7 @@ public struct Foo : Differentiable {
 // CHECK-AST:   public typealias CotangentVector = Foo.AllDifferentiableVariables
 
 // CHECK-SILGEN-LABEL: // Foo.a.getter
-// CHECK-SILGEN-NEXT: sil [transparent] [serialized] [differentiable source 0 wrt 0] @$s33derived_differentiable_properties3FooV1aSfvg : $@convention(method) (Foo) -> Float
+// CHECK-SILGEN-NEXT: sil [transparent] [serialized] [differentiable source 0 wrt 0] [ossa] @$s33derived_differentiable_properties3FooV1aSfvg : $@convention(method) (Foo) -> Float
 
 struct AdditiveTangentIsSelf : AdditiveArithmetic, Differentiable {
   var a: Float

test/AutoDiff/differentiable_attr_silgen.swift

@@ -10,14 +10,14 @@ public func foo(_ x: Float, _ y: Float) -> Float {
   return 1
 }
 
-// CHECK-LABEL: sil [differentiable source 0 wrt 0, 1 vjp @dfoo] @foo
+// CHECK-LABEL: sil [differentiable source 0 wrt 0, 1 vjp @dfoo] [ossa] @foo
 
 @_silgen_name("dfoo")
 public func dfoo(_ x: Float, _ y: Float) -> (Float, (Float) -> (Float, Float)) {
   return (foo(x, y), { _ in (1, 1) })
 }
 
-// CHECK-LABEL: sil @dfoo
+// CHECK-LABEL: sil [ossa] @dfoo
 
 //===----------------------------------------------------------------------===//
 // Indirect returns
@@ -29,8 +29,8 @@ public func foo_indir_ret<T: Differentiable>(_ x: Float, _ y: T) -> T {
   return y
 }
 
-// CHECK-LABEL: sil [differentiable source 0 wrt 0, 1 vjp @dfoo_indir_ret] @foo_indir_ret : $@convention(thin) <T where T : Differentiable> (Float, @in_guaranteed T) -> @out T {
-// CHECK: bb0(%0 : @trivial $*T, %1 : @trivial $Float, %2 : @trivial $*T):
+// CHECK-LABEL: sil [differentiable source 0 wrt 0, 1 vjp @dfoo_indir_ret] [ossa] @foo_indir_ret : $@convention(thin) <T where T : Differentiable> (Float, @in_guaranteed T) -> @out T {
+// CHECK: bb0(%0 : $*T, %1 : $Float, %2 : $*T):
 
 @_silgen_name("dfoo_indir_ret")
 public func dfoo_indir_ret<T: Differentiable>(_ x: Float, _ y: T) -> (T, (T.CotangentVector) -> (Float, T.CotangentVector)) {
@@ -47,14 +47,14 @@ public func hasjvp(_ x: Float, _ y: Float) -> Float {
   return 1
 }
 
-// CHECK-LABEL: sil [differentiable source 0 wrt 0, 1 jvp @dhasjvp] @hasjvp
+// CHECK-LABEL: sil [differentiable source 0 wrt 0, 1 jvp @dhasjvp] [ossa] @hasjvp
 
 @_silgen_name("dhasjvp")
 public func dhasjvp(_ x: Float, _ y: Float) -> (Float, (Float, Float) -> Float) {
   return (1, { _, _ in 1 })
 }
 
-// CHECK-LABEL: sil @dhasjvp
+// CHECK-LABEL: sil [ossa] @dhasjvp
 
 //===----------------------------------------------------------------------===//
 // VJP
@@ -67,14 +67,14 @@ public func hasvjp(_ x: Float, _ y: Float) -> Float {
   return 1
 }
 
-// CHECK-LABEL: sil [serialized] [differentiable source 0 wrt 0, 1 vjp @dhasvjp] @hasvjp
+// CHECK-LABEL: sil [serialized] [differentiable source 0 wrt 0, 1 vjp @dhasvjp] [ossa] @hasvjp
 
 @_silgen_name("dhasvjp")
 public func dhasvjp(_ x: Float, _ y: Float) -> (Float, (Float) -> (Float, Float)) {
   return (1, { _ in (1, 1) })
 }
 
-// CHECK-LABEL: sil @dhasvjp
+// CHECK-LABEL: sil [ossa] @dhasvjp
 
 //===----------------------------------------------------------------------===//
 // Stored property
@@ -163,4 +163,4 @@ public struct MyLayer: Differentiable {
 }
 
 // CHECK-LABEL: initialization expression of MyLayer.x
-// CHECK-NEXT: sil [transparent] @$s26differentiable_attr_silgen7MyLayerV1xSfvpfi : $@convention(thin) () -> Float
+// CHECK-NEXT: sil [transparent] [ossa] @$s26differentiable_attr_silgen7MyLayerV1xSfvpfi : $@convention(thin) () -> Float

test/AutoDiff/differentiable_attr_type_checking.swift

@@ -564,8 +564,9 @@ func vjpNonvariadic(_ x: Float, indices: [Int32]) -> (Float, (Float) -> Float) {
   return (x, { $0 })
 }
 
-// expected-error @+2 {{type 'Scalar' constrained to non-protocol, non-class type 'Float'}}
-// expected-error @+1 {{can only differentiate with respect to parameters that conform to 'Differentiable', but 'Scalar' does not conform to 'Differentiable'}}
+// expected-error @+3 {{type 'Scalar' constrained to non-protocol, non-class type 'Float'}}
+// expected-error @+2 {{can only differentiate with respect to parameters that conform to 'Differentiable', but 'Scalar' does not conform to 'Differentiable'}}
+// expected-note @+1 {{use 'Scalar == Float' to require 'Scalar' to be 'Float'}}
 @differentiable(where Scalar : Float)
 func invalidRequirementConformance<Scalar>(x: Scalar) -> Scalar {
   return x