averello/swift-mirror
1
0
Fork 0
mirror of https://github.com/apple/swift.git synced 2025-12-21 12:14:44 +01:00
Code Issues Packages Projects Releases Wiki Activity

[AutoDiff] Start linear_function canonicalization skeleton (#33057)

Browse Source 
Start `linear_function` canonicalization skeleton copying from
`differentiable_function` canonicalization. For now, transpose function
operands are filled in with `undef`.
This commit is contained in:
Alex Efremov
2020-07-23 23:10:50 +02:00
committed by GitHub
parent cde03e66e3
commit 1a81573eed
8 changed files with 244 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
test/AutoDiff/SILOptimizer/linear_function_canonicalization.sil Normal file
View File

@@ -0,0 +1,28 @@
// RUN: %target-sil-opt -differentiation -enable-experimental-linear-map-transposition %s | %FileCheck %s
sil_stage raw
import Swift
import Builtin
import _Differentiation
sil hidden @foo : $@convention(thin) (Float, Float, Float) -> Float {
bb0(%0 : $Float, %1 : $Float, %2 : $Float):
return %2 : $Float
}
sil @make_linear_func : $@convention(thin) () -> () {
bb0:
%orig = function_ref @foo : $@convention(thin) (Float, Float, Float) -> Float
%linear_fn_0 = linear_function [parameters 0] %orig : $@convention(thin) (Float, Float, Float) -> Float
%linear_fn_1 = linear_function [parameters 0 2] %orig : $@convention(thin) (Float, Float, Float) -> Float
return undef : $()
}
// CHECK-LABEL: sil @make_linear_func
// CHECK: bb0:
// CHECK: [[ORIG_FN:%.*]] = function_ref @foo : $@convention(thin) (Float, Float, Float) -> Float
// CHECK: linear_function [parameters 0] %0 : $@convention(thin) (Float, Float, Float) -> Float with_transpose undef : $@convention(thin) (Float, Float, Float) -> Float
// CHECK: linear_function [parameters 0 2] %0 : $@convention(thin) (Float, Float, Float) -> Float with_transpose undef : $@convention(thin) (Float, Float) -> (Float, Float)
// CHECK: }