File tree Expand file tree Collapse file tree 1 file changed +42
-0
lines changed Expand file tree Collapse file tree 1 file changed +42
-0
lines changed Original file line number Diff line number Diff line change 1+ //@ compile-flags: -Zautodiff=Enable -Zautodiff=NoPostopt -C opt-level=3 -Clto=fat
2+ //@ no-prefer-dynamic
3+ //@ needs-enzyme
4+ #![ feature( autodiff) ]
5+
6+ use std:: autodiff:: autodiff;
7+
8+ #[ autodiff( d_square, Reverse , Duplicated , Active ) ]
9+ fn square < T : std:: ops:: Mul < Output = T > + Copy > ( x : & T ) -> T {
10+ * x * * x
11+ }
12+
13+ // Ensure that `d_square::<f64>` code is generated even if `square::<f64>` was never called
14+ //
15+ // CHECK: ; generic::square
16+ // CHECK-NEXT: ; Function Attrs:
17+ // CHECK-NEXT: define internal {{.*}} double
18+ // CHECK-NEXT: start:
19+ // CHECK-NOT: ret
20+ // CHECK: fmul double
21+
22+ // Ensure that `d_square::<f32>` code is generated
23+ //
24+ // CHECK: ; generic::square
25+ // CHECK-NEXT: ; Function Attrs: {{.*}}
26+ // CHECK-NEXT: define internal {{.*}} float
27+ // CHECK-NEXT: start:
28+ // CHECK-NOT: ret
29+ // CHECK: fmul float
30+
31+ fn main ( ) {
32+ let xf32: f32 = std:: hint:: black_box ( 3.0 ) ;
33+ let xf64: f64 = std:: hint:: black_box ( 3.0 ) ;
34+
35+ let outputf32 = square :: < f32 > ( & xf32) ;
36+ assert_eq ! ( 9.0 , outputf32) ;
37+
38+ let mut df_dxf64: f64 = std:: hint:: black_box ( 0.0 ) ;
39+
40+ let output_f64 = d_square :: < f64 > ( & xf64, & mut df_dxf64, 1.0 ) ;
41+ assert_eq ! ( 6.0 , df_dxf64) ;
42+ }
You can’t perform that action at this time.
0 commit comments