Skip to content

implement dpnp.fft.rfft and dpnp.fft.irfft #1928

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
vtavana merged 11 commits into from
Jul 30, 2024
Merged

implement dpnp.fft.rfft and dpnp.fft.irfft #1928

Changes from 1 commit
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
b37bb2d
implement rfft and irfft
Jun 29, 2024
39a27b5
Merge branch 'master' into rfft
vtavana Jul 15, 2024
92d4070
update tests
Jul 15, 2024
d8b4856
remove redundant test
Jul 22, 2024
d3bd05a
clean-up
Jul 22, 2024
b72de00
separate backend structures
Jul 22, 2024
c0630a2
address comments
Jul 22, 2024
669168d
Merge branch 'master' into rfft
vtavana Jul 22, 2024
6c17c09
mute tests on gpu with fp64 support
Jul 26, 2024
dc30a05
Merge branch 'master' into rfft
vtavana Jul 26, 2024
4d8a64a
Merge branch 'master' into rfft
vtavana Jul 29, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
remove redundant test
  • Loading branch information
Vahid Tavanashad committed Jul 22, 2024
commit d8b4856780510e42fa503ab36225125abc186f1f
24 changes: 0 additions & 24 deletions tests/test_sycl_queue.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1231,34 +1231,10 @@ def test_fft(func, device):

expected = getattr(numpy.fft, func)(data)
result = getattr(dpnp.fft, func)(dpnp_data)

assert_dtype_allclose(result, expected)

expected_queue = dpnp_data.get_array().sycl_queue
result_queue = result.get_array().sycl_queue

assert_sycl_queue_equal(result_queue, expected_queue)


@pytest.mark.parametrize("type", ["float32"])
@pytest.mark.parametrize("shape", [(8, 8)])
@pytest.mark.parametrize(
"device",
valid_devices,
ids=[device.filter_string for device in valid_devices],
)
def test_fft_rfft(type, shape, device):
np_data = numpy.arange(64, dtype=numpy.dtype(type)).reshape(shape)
dpnp_data = dpnp.array(np_data, device=device)

np_res = numpy.fft.rfft(np_data)
dpnp_res = dpnp.fft.rfft(dpnp_data)

assert_dtype_allclose(dpnp_res, np_res, check_only_type_kind=True)

expected_queue = dpnp_data.get_array().sycl_queue
result_queue = dpnp_res.get_array().sycl_queue

assert_sycl_queue_equal(result_queue, expected_queue)


Expand Down