| // RUN: %empty-directory(%t) |
| // RUN: %target-swift-frontend -emit-module %s -o %t/autodiff_generated_decl_member_loading_cross_module.swiftmodule |
| // RUN: %target-swift-frontend -merge-modules -emit-module %t/autodiff_generated_decl_member_loading_cross_module.swiftmodule |
| |
| // Tests TF-805. |
| // |
| // Previously, `IterableDeclContext::loadAllMembers` was disabled for |
| // AD-generated structs/enums: |
| // https://github.com/apple/swift/commit/7e89bee39bfda4624f04dcc2c8d53599fbde6191 |
| // |
| // This caused a SIL verification failure for because enum members were not loaded |
| // when running `-emit-module` then `-merge-modules -sil-merge-partial-modules`: |
| // |
| // SIL verification failed: switch_enum dispatches on same enum element |
| // more than once: unswitchedElts.count(elt) |
| |
| public struct Tensor<Scalar> {} |
| |
| extension Tensor: Differentiable where Scalar: Differentiable{ |
| public typealias TangentVector = Float |
| public mutating func move(along direction: Float) {} |
| } |
| |
| extension Tensor { |
| @inlinable |
| @differentiable(wrt: self where Scalar: Differentiable) |
| func TF_805(axis: Int) -> Tensor { |
| if axis != axis {} |
| return self |
| } |
| } |
