fix safe fft performance

EnzymeAD · Oct 2, 2024 · 1fe64c0 · 1fe64c0
1 parent 05d3e87
commit 1fe64c0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/enzyme/benchmarks/ReverseMode/fft/src/lib.rs b/enzyme/benchmarks/ReverseMode/fft/src/lib.rs
@@ -1,3 +1,4 @@
+#![feature(slice_swap_unchecked)]
 #![feature(autodiff)]
 
 pub mod safe;

diff --git a/enzyme/benchmarks/ReverseMode/fft/src/safe.rs b/enzyme/benchmarks/ReverseMode/fft/src/safe.rs
@@ -9,8 +9,8 @@ fn bitreversal_perm<T>(data: &mut [T]) {
  while i < 2*len {
  if j > i {
  //dbg!(&i, &j);
- data.swap(j-1, i-1);
- data.swap(j, i);
+ unsafe {data.swap_unchecked(j-1, i-1);}
+ unsafe {data.swap_unchecked(j, i);}
  }
 
  let mut m = len;