mirror of
https://github.com/rust-lang/rust.git
synced 2025-10-02 18:27:37 +00:00
31 lines
610 B
Rust
31 lines
610 B
Rust
//@ compile-flags: -Zautodiff=Enable -Zautodiff=NoPostopt -C opt-level=3 -Clto=fat
|
|
//@ no-prefer-dynamic
|
|
//@ needs-enzyme
|
|
|
|
// Just check it does not crash for now
|
|
// CHECK: ;
|
|
#![feature(autodiff)]
|
|
|
|
use std::autodiff::autodiff_reverse;
|
|
|
|
struct Foo {
|
|
a: f64,
|
|
}
|
|
|
|
trait MyTrait {
|
|
fn f(&self, x: f64) -> f64;
|
|
fn df(&self, x: f64, seed: f64) -> (f64, f64);
|
|
}
|
|
|
|
impl MyTrait for Foo {
|
|
#[autodiff_reverse(df, Const, Active, Active)]
|
|
fn f(&self, x: f64) -> f64 {
|
|
self.a * 0.25 * (x * x - 1.0 - 2.0 * x.ln())
|
|
}
|
|
}
|
|
|
|
fn main() {
|
|
let foo = Foo { a: 3.0f64 };
|
|
dbg!(foo.df(1.0, 1.0));
|
|
}
|