Skip to content

Commit e344b3d

Browse files
authored
Fix wrong derivative mode in test (rust-lang#382)
* Fix wrong derivative mode * add missing function to tests
1 parent 6764d51 commit e344b3d

File tree

2 files changed

+11
-2
lines changed

2 files changed

+11
-2
lines changed

enzyme/test/Enzyme/ForwardMode/badcall4.ll

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,4 +78,13 @@ declare dso_local double @__enzyme_fwddiff(i8*, double*, double*) local_unnamed_
7878
; CHECK-NEXT: store double 3.000000e+00, double* %arrayidx, align 8
7979
; CHECK-NEXT: store double 0.000000e+00, double* %"arrayidx'ipg", align 8
8080
; CHECK-NEXT: ret void
81+
; CHECK-NEXT: }
82+
83+
; CHECK: define internal {{(dso_local )?}}void @fwddiffeothermetasubf(double* nocapture %x, double* nocapture %"x'")
84+
; CHECK-NEXT: entry:
85+
; CHECK-NEXT: %"arrayidx'ipg" = getelementptr inbounds double, double* %"x'", i64 1
86+
; CHECK-NEXT: %arrayidx = getelementptr inbounds double, double* %x, i64 1
87+
; CHECK-NEXT: store double 4.000000e+00, double* %arrayidx, align 8
88+
; CHECK-NEXT: store double 0.000000e+00, double* %"arrayidx'ipg", align 8
89+
; CHECK-NEXT: ret void
8190
; CHECK-NEXT: }

enzyme/test/Enzyme/ForwardMode/sin.ll

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ entry:
99

1010
define double @test_derivative(double %x) {
1111
entry:
12-
%0 = tail call double (double (double)*, ...) @__enzyme_autodiff(double (double)* nonnull @tester, double %x)
12+
%0 = tail call double (double (double)*, ...) @__enzyme_fwddiff(double (double)* nonnull @tester, double %x, double 1.0)
1313
ret double %0
1414
}
1515

@@ -20,7 +20,7 @@ declare double @llvm.cos.f64(double)
2020
declare double @llvm.sin.f64(double)
2121

2222
; Function Attrs: nounwind
23-
declare double @__enzyme_autodiff(double (double)*, ...)
23+
declare double @__enzyme_fwddiff(double (double)*, ...)
2424

2525
; CHECK: define double @test_derivative(double %x)
2626
; CHECK-NEXT: entry:

0 commit comments

Comments
 (0)