adding rust ADBench version #1797
Conversation
|
Usage instruction: Enzyme/enzyme/Enzyme/GradientUtils.cpp Line 9136 in 880fb32 |
|
printing the module before AD (with a few helping infos at the end) |
| @@ -127,6 +127,19 @@ extern "C" { | |||
| double* reproj_err, | |||
| double* w_err | |||
| ); | |||
|
|
|||
| void rust2_ba_objective( | |||
There was a problem hiding this comment.
weak symbol variant of this.
-> running these doesn't require rust.
| %-opt.ll: %-raw.ll | ||
| opt $^ -o $@ -S | ||
| #opt $^ -O2 -o $@ -S | ||
| $(dir)/benchmarks/ReverseMode/ba/target/release/libbars.a: src/lib.rs Cargo.toml |
There was a problem hiding this comment.
What would you like this variable to use when determining if Rust is enabled? We could run cargo +enzyme version (checking whether a toolchain named enzyme is configured) and/or we could have an environment variable to be set (by CMake or otherwise).
| fn radix2(data: &mut [f64], i_sign: f64) { | ||
| let n = data.len() / 2; | ||
| if n == 1 { | ||
| return; | ||
| } | ||
|
|
||
| let (a,b) = data.split_at_mut(n); | ||
| radix2(a, i_sign, n/2); | ||
| radix2(b, i_sign, n/2); | ||
| let (a, b) = data.split_at_mut(n); | ||
| radix2(a, i_sign); | ||
| radix2(b, i_sign); | ||
|
|
||
| let wtemp = i_sign * (PI / n as f64).sin(); | ||
| let wpi = -i_sign * (2.0 * PI / n as f64).sin(); | ||
| let wpr = -2.0 * wtemp * wtemp; | ||
| let mut wr = 1.0; | ||
| let mut wi = 0.0; | ||
|
|
||
| assert_eq!(data.len(), 2*n); | ||
| for i in (0..n).step_by(2) { | ||
| let in_n = i + n; | ||
| let tempr = data[in_n] * wr - data[in_n + 1] * wi; | ||
| let tempi = data[in_n] * wi + data[in_n + 1] * wr; | ||
| for i in (0..n).step_by(2) { | ||
| let tempr = b[i] * wr - b[i + 1] * wi; | ||
| let tempi = b[i] * wi + b[i + 1] * wr; | ||
|
|
||
| data[in_n] = data[i] - tempr; | ||
| data[in_n + 1] = data[i + 1] - tempi; | ||
| data[i] += tempr; | ||
| data[i + 1] += tempi; | ||
| b[i] = a[i] - tempr; | ||
| b[i + 1] = a[i + 1] - tempi; | ||
| a[i] += tempr; | ||
| a[i + 1] += tempi; | ||
|
|
||
| let wtemp_new = wr; | ||
| wr += wr * wpr - wi * wpi; | ||
| wi += wi * wpr + wtemp_new * wpi; | ||
| } | ||
|
|
||
| } |
There was a problem hiding this comment.
@ZuseZ4 Let me know if you don't like this and I'll remove it. (Other changes in this commit are just rustfmt/whitespace.)
* main: (49 commits) Fix iv of constant (#2141) Update benchmarks (#2035) Implement tgamma derivative (#2140) tgamma error improvement (#2139) Improve cache index error message (#2138) Fixes warnings and adds missing header guards (#2124) mlir: cache and reuse reverse funcs (#2133) mlir: implement forward mode for func.call (#2134) mlir: Func call reverse diff (#2127) Update build_tarballs.jl Fix combined temp cache for reverse (#2131) Improve runtime activity err message (#2132) Fix undef value storage (#2129) Adapt to const tblgen (#2128) Add gcloaded TT (#2125) Fix blas decl updater indexing (#2123) Add header files to ClangEnzyme target (#2062) Improve unknown function error messages (#2120) Fix handle sync (#2122) Support more Julia 1.11 functions (#2121) ...
This reverts commit 3b174ea.
This yields identical assembly apart from asserts at the top of safe::radix2.
Note that -mtune=cascadelake is only about tuning (like instruction timing) not arch/features.
|
closing as discussed, since maintaining this benchmark infra turned out to be too much work and this has no outlook of being merged. |
Usage for Rust side
cargo +enzyme run --releaseComipling a lib.so to use from C:
cargo +enzyme rustc --release --lib --crate-type=cdylib