File: tf1315-pullback-subset-parameter-thunk-generation.swift

package info (click to toggle)
swiftlang 6.0.3-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 2,519,992 kB
  • sloc: cpp: 9,107,863; ansic: 2,040,022; asm: 1,135,751; python: 296,500; objc: 82,456; f90: 60,502; lisp: 34,951; pascal: 19,946; sh: 18,133; perl: 7,482; ml: 4,937; javascript: 4,117; makefile: 3,840; awk: 3,535; xml: 914; fortran: 619; cs: 573; ruby: 573
file content (26 lines) | stat: -rw-r--r-- 743 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
// RUN: %target-swift-frontend -emit-sil %s

// TF-1315: Pullback subset thunk generation crash due to unmapped parameter
// index for `inout` differentiability parameters.

import _Differentiation

func foo(_ x: Int, _ y: Float, _ z: inout Float) {}

@derivative(of: foo, wrt: (y, z))
func vjpFoo(_ x: Int, _ y: Float, _ z: inout Float) -> (
  value: Void, pullback: (inout Float) -> Float
) {
  fatalError()
}

@differentiable(reverse)
func TF_1315(_ x: Float) -> Float {
  var x = x
  // The call to `foo` below triggers pullback subset parameter thunk generation.
  // `foo` original function type: `(Int, Float, inout Float) -> ()`
  //     Actual parameter indices: 1, 2
  //    Desired parameter indices: 2
  foo(1, 2, &x)
  return x
}