| // RUN: %target-build-swift -O %s |
| |
| // SR-12493: SIL verification error regarding substituted function types and |
| // `differentiable_function_extract` instruction. Occurs only with `-O`. |
| |
| // SWIFT_ENABLE_TENSORFLOW |
| // Note: SR-12493 occurs only on master branch, not on tensorflow branch. |
| // UNSUPPORTED: tensorflow |
| // SWIFT_ENABLE_TENSORFLOW END |
| |
| // FIXME(SR-13021): Disabled due to flakiness on Linux, likely related to TF-1197. |
| // REQUIRES: SR13021 |
| |
| import _Differentiation |
| |
| func exampleVJP_1(_ x0: Float) -> (Float, (Float) -> (Float)) { |
| ( |
| x0 * x0, |
| { (2 * x0 * $0) } |
| ) |
| } |
| |
| func bar() { |
| let f = differentiableFunction(from: exampleVJP_1) |
| let pb = pullback(at: 10, in: f) |
| _ = pb(1) |
| } |
