Skip to content

Commit

Permalink
set strides, distance and offset on the basis of direction
Browse files Browse the repository at this point in the history
  • Loading branch information
AD2605 committed Jan 19, 2024
1 parent 93ab8a4 commit 6fcb7d4
Showing 1 changed file with 8 additions and 3 deletions.
11 changes: 8 additions & 3 deletions src/portfft/descriptor.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -1186,9 +1186,14 @@ class committed_descriptor {
"To use interface with interleaved real and imaginary values, descriptor.complex_storage must be set to "
"INTERLEAVED_COMPLEX.");
}
return dispatch_dimensions(in, out, in_imag, out_imag, dependencies, params.forward_strides,
params.backward_strides, params.forward_distance, params.backward_distance,
params.forward_offset, params.backward_offset, compute_direction);
if (compute_direction == direction::FORWARD) {
return dispatch_dimensions(in, out, in_imag, out_imag, dependencies, params.forward_strides,
params.backward_strides, params.forward_distance, params.backward_distance,
params.forward_offset, params.backward_offset, compute_direction);
}
return dispatch_dimensions(in, out, in_imag, out_imag, dependencies, params.backward_strides,
params.forward_strides, params.backward_distance, params.forward_distance,
params.backward_offset, params.forward_offset, compute_direction);
}

/**
Expand Down

0 comments on commit 6fcb7d4

Please sign in to comment.