Update generic tests
This commit is contained in:
@@ -3,9 +3,9 @@
|
||||
//@ needs-enzyme
|
||||
#![feature(autodiff)]
|
||||
|
||||
use std::autodiff::autodiff;
|
||||
use std::autodiff::autodiff_reverse;
|
||||
|
||||
#[autodiff(d_square, Reverse, Duplicated, Active)]
|
||||
#[autodiff_reverse(d_square, Duplicated, Active)]
|
||||
fn square<T: std::ops::Mul<Output = T> + Copy>(x: &T) -> T {
|
||||
*x * *x
|
||||
}
|
||||
|
||||
@@ -64,7 +64,7 @@ pub fn f9() {
|
||||
}
|
||||
|
||||
// Make sure we can handle generics
|
||||
#[autodiff(d_square, Reverse, Duplicated, Active)]
|
||||
#[autodiff_reverse(d_square, Duplicated, Active)]
|
||||
pub fn f10<T: std::ops::Mul<Output = T> + Copy>(x: &T) -> T {
|
||||
*x * *x
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user