Skip to content

Commit b3d9e93

Browse files
committed
[computed_tomography] updating computed_tomography to recommended usage
1 parent 1f4539f commit b3d9e93

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

Libraries/oneMKL/computed_tomography/computed_tomography.cpp

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -288,8 +288,11 @@ sycl::event step3_ifft_2d(matrix_r &fhat,
288288
const std::vector<sycl::event> &deps)
289289
{
290290
// Configure descriptor
291-
std::int64_t strides[3] = {0, (fhat.ldw) / 2, 1}; // fhat.ldw/2, in complex'es
292-
ifft2d.set_value(oneapi::mkl::dft::config_param::INPUT_STRIDES, strides);
291+
std::vector<std::int64_t> strides = {0, (fhat.ldw) / 2, 1}; // fhat.ldw/2, in complex'es
292+
// Strides must be identical in forward and backward domain for in-place
293+
// complex transforms
294+
ifft2d.set_value(oneapi::mkl::dft::config_param::FWD_STRIDES, strides);
295+
ifft2d.set_value(oneapi::mkl::dft::config_param::BWD_STRIDES, strides);
293296
ifft2d.commit(main_queue);
294297

295298
sycl::event ifft2d_ev = oneapi::mkl::dft::compute_backward(ifft2d, fhat.data, deps);

0 commit comments

Comments
 (0)