mirror of
https://github.com/rust-lang/rust.git
synced 2026-04-27 18:57:42 +03:00
Manuel Drehwald
26c9f7255a
It will use enzyme to generate a default derivative implementation, which can be overwritten by the user.
33 lines
601 B
Rust
33 lines
601 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)]
|
|
#![feature(core_intrinsics)]
|
|
#![feature(rustc_attrs)]
|
|
|
|
use std::autodiff::autodiff_reverse;
|
|
|
|
struct Foo {
|
|
a: f64,
|
|
}
|
|
|
|
trait MyTrait {
|
|
#[autodiff_reverse(df, Const, Active, Active)]
|
|
fn f(&self, x: f64) -> f64;
|
|
}
|
|
|
|
impl MyTrait for Foo {
|
|
fn f(&self, x: f64) -> f64 {
|
|
x.sin()
|
|
}
|
|
}
|
|
|
|
fn main() {
|
|
let foo = Foo { a: 3.0f64 };
|
|
dbg!(foo.df(2.0, 1.0));
|
|
dbg!(2.0_f64.cos());
|
|
}
|