mirror of
https://github.com/apple/swift.git
synced 2025-12-14 20:36:38 +01:00
198d4ab0bb
Introduce a new pass MandatoryTempRValueElimination, which works as the original TempRValueElimination, except that it does not remove any alloc_stack instruction which are associated with source variables. Running this pass at Onone helps to reduce copies of large structs, e.g. InlineArrays or structs containing InlineArrays. Copying large structs can be a performance problem, even at Onone. rdar://151629149
41 lines
1.9 KiB
Swift
41 lines
1.9 KiB
Swift
// RUN: %target-swift-emit-sil -Xllvm -sil-print-types %s | %FileCheck %s
|
|
|
|
import _Differentiation
|
|
|
|
// CHECK: sil private{{.*}}@$s17optional_pullback23givesWrongTangentVector1xxSgx_t16_Differentiation14DifferentiableRzlFAeFRzlTJpSpSr
|
|
// CHECK-SAME: $@convention(thin) <τ_0_0 where τ_0_0 : Differentiable>
|
|
// CHECK-SAME: (@in_guaranteed Optional<τ_0_0>.TangentVector) -> @out τ_0_0.TangentVector
|
|
//
|
|
// CHECK: bb0(%[[RET_TAN:.+]] : $*τ_0_0.TangentVector, %[[OPT_TAN:.+]] : $*Optional<τ_0_0>.TangentVector):
|
|
// CHECK: %[[RET_TAN_BUF:.+]] = alloc_stack $τ_0_0.TangentVector, let, name "derivative of 'x'
|
|
|
|
// CHECK: %[[ZERO1:.+]] = witness_method $τ_0_0.TangentVector, #AdditiveArithmetic.zero!getter
|
|
// CHECK: apply %[[ZERO1]]<τ_0_0.TangentVector>(%[[RET_TAN_BUF]], %{{.*}})
|
|
// CHECK: %[[ADJ_IN_BB:.+]] = alloc_stack $τ_0_0.TangentVector
|
|
//
|
|
// CHECK: %[[TAN_VAL:.+]] = struct_element_addr %[[OPT_TAN]] : $*Optional<τ_0_0>.TangentVector, #Optional.TangentVector.value
|
|
//
|
|
// CHECK: %[[TAN_DATA:.+]] = unchecked_take_enum_data_addr %[[TAN_VAL]] : $*Optional<τ_0_0.TangentVector>, #Optional.some!enumelt
|
|
// CHECK: %[[PLUS_EQUAL:.+]] = witness_method $τ_0_0.TangentVector, #AdditiveArithmetic."+="
|
|
// CHECK: apply %[[PLUS_EQUAL]]<τ_0_0.TangentVector>(%[[ADJ_IN_BB]], %[[TAN_DATA]], %{{.*}})
|
|
|
|
// CHECK: %[[PLUS_EQUAL:.+]] = witness_method $τ_0_0.TangentVector, #AdditiveArithmetic."+="
|
|
// CHECK: apply %[[PLUS_EQUAL]]<τ_0_0.TangentVector>(%[[RET_TAN_BUF]], %[[ADJ_IN_BB]], %{{.*}})
|
|
// CHECK: destroy_addr %[[ADJ_IN_BB]] : $*τ_0_0.TangentVector
|
|
|
|
// CHECK: copy_addr [take] %[[RET_TAN_BUF:.+]] to [init] %[[RET_TAN:.+]]
|
|
// CHECK: dealloc_stack %[[RET_TAN_BUF]] : $*τ_0_0.TangentVector
|
|
|
|
@differentiable(reverse)
|
|
func givesWrongTangentVector<Element>(x: Element) -> Element? where Element: Differentiable {
|
|
return x
|
|
}
|
|
|
|
@differentiable(reverse)
|
|
func f(x: Double) -> Double {
|
|
let y = givesWrongTangentVector(x: x)
|
|
return y!
|
|
}
|
|
|
|
print(valueWithGradient(at: 0.0, of: f))
|