Apply changes to tests in #27659 and add IRGen RUN line

Author: rxwei

test/AutoDiff/differentiable_func_type.sil

@@ -0,0 +1,31 @@
+// RUN: %empty-directory(%t)
+// RUN: %target-sil-opt %s -emit-sib -o %t/tmp.sib -module-name differentiable_func_type
+// RUN: %target-sil-opt %t/tmp.sib -o %t/tmp.2.sib -module-name differentiable_func_type
+// RUN: %target-sil-opt %t/tmp.2.sib -module-name differentiable_func_type | %FileCheck %s -check-prefix=CHECK-SIL
+
+// RUN: %target-swift-frontend %s -emit-ir -module-name differentiable_func_type | %FileCheck %s -check-prefix=CHECK-LLVM
+
+sil_stage raw
+
+import Swift
+
+sil @takeAndReturnLinear : $@convention(thin) (@differentiable(linear) (Float) -> Float) -> @differentiable(linear) (Float) -> Float {
+bb0(%0 : $@differentiable(linear) (Float) -> Float):
+  return %0 : $@differentiable(linear) (Float) -> Float
+}
+
+// CHECK-LABEL: sil @takeAndReturnLinear : $@convention(thin) (@differentiable(linear) (Float) -> Float) -> @differentiable(linear) (Float) -> Float {
+// CHECK: bb0([[ARG:%.*]] : $@differentiable(linear) (Float) -> Float):
+// CHECK:   return [[ARG]] : $@differentiable(linear) (Float) -> Float
+// CHECK: }
+
+
+sil @takeAndReturnDifferentiable : $@convention(thin) (@differentiable (Float) -> Float) -> @differentiable (Float) -> Float {
+bb0(%0 : $@differentiable (Float) -> Float):
+  return %0 : $@differentiable (Float) -> Float
+}
+
+// CHECK-LABEL: sil @takeAndReturnDifferentiable : $@convention(thin) (@differentiable (Float) -> Float) -> @differentiable (Float) -> Float {
+// CHECK: bb0([[ARG:%.*]] : $@differentiable (Float) -> Float):
+// CHECK:   return [[ARG]] : $@differentiable (Float) -> Float
+// CHECK: }

test/AutoDiff/differentiable_function_inst.sil

@@ -10,6 +10,35 @@ sil_stage raw
 import Swift
 import Builtin
 
+sil @examplefunc : $@convention(thin) (Float, Float, Float) -> Float
+sil @examplemethod : $@convention(method) (Float, Float, Float) -> Float
+
+// CHECK-LABEL: sil @test
+sil @test : $@convention(thin) () -> () {
+bb0:
+  %0 = function_ref @examplefunc : $@convention(thin) (Float, Float, Float) -> Float
+  %1 = differentiable_function [wrt 0 1 2] %0 : $@convention(thin) (Float, Float, Float) -> Float
+
+  // CHECK: %2 = differentiable_function_extract [vjp] %1 : $@differentiable @convention(thin) (Float, Float, Float) -> Float
+  %2 = differentiable_function_extract [vjp] %1 : $@differentiable @convention(thin) (Float, Float, Float) -> Float
+  %3 = differentiable_function [wrt 0] %0 : $@convention(thin) (Float, Float, Float) -> Float
+
+  // CHECK: %4 = differentiable_function_extract [vjp] %3 : $@differentiable @convention(thin) (Float, @nondiff Float, @nondiff Float) -> Float
+  %4 = differentiable_function_extract [vjp] %3 : $@differentiable @convention(thin) (Float, @nondiff Float, @nondiff Float) -> Float
+  %5 = function_ref @examplemethod : $@convention(method) (Float, Float, Float) -> Float
+  %6 = differentiable_function [wrt 0 1 2] %5 : $@convention(method) (Float, Float, Float) -> Float
+
+  // CHECK: %7 = differentiable_function_extract [vjp] %6 : $@differentiable @convention(method) (Float, Float, Float) -> Float
+  %7 = differentiable_function_extract [vjp] %6 : $@differentiable @convention(method) (Float, Float, Float) -> Float
+  %8 = differentiable_function [wrt 0] %5 : $@convention(method) (Float, Float, Float) -> Float
+
+  // CHECK: %9 = differentiable_function_extract [vjp] %8 : $@differentiable @convention(method) (Float, @nondiff Float, @nondiff Float) -> Float
+  %9 = differentiable_function_extract [vjp] %8 : $@differentiable @convention(method) (Float, @nondiff Float, @nondiff Float) -> Float
+
+  %ret = tuple ()
+  return %ret : $()
+}
+
 // The adjoint function emitted by the compiler. Parameter are a vector, as in
 // vector-Jacobian products, and pullback values. The function is partially
 // applied to a pullback struct to form a pullback, which takes a vector and