Skip to content

Fix incorrect results from cumulative_sum and cumulative_prod with dtype=bool #2018

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
ndgrigorian merged 6 commits into from
Feb 28, 2025
Merged

Fix incorrect results from cumulative_sum and cumulative_prod with dtype=bool #2018

Changes from 1 commit
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
e2d2b3f
Add missing kernels to arithmetic accumulators
ndgrigorian Feb 25, 2025
47239cd
Add tests based on gh-2017
ndgrigorian Feb 25, 2025
48b0e66
Add comment specifying need for work-around in accumulators
ndgrigorian Feb 26, 2025
c58d775
Reuse SYCL utils in workaround for logical operators in accumulators
ndgrigorian Feb 26, 2025
35d46d3
Alias conditional binary op type for cumulative_sum and cumulative_prod
ndgrigorian Feb 26, 2025
fa525a7
Add comment to needs_workaround in accumulators.hpp
ndgrigorian Feb 28, 2025
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
Add tests based on gh-2017
  • Loading branch information
@ndgrigorian
ndgrigorian committed Feb 25, 2025
commit 47239cdd6f01bb09e2f010eb82ce2a56c3253776
12 changes: 12 additions & 0 deletions dpctl/tests/test_tensor_accumulation.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -421,3 +421,15 @@ def test_cumulative_sum_gh_1901(p):
inp = dpt.ones(n, dtype=dt)
r = dpt.cumulative_sum(inp, dtype=dt)
assert dpt.all(r == dpt.arange(1, n + 1, dtype=dt))


@pytest.mark.parametrize(
"dt", ["i1", "i2", "i4", "i8", "f2", "f4", "f8", "c8", "c16"]
)
def test_gh_2017(dt):
"See https://github.com/IntelPython/dpctl/issues/2017"
q = get_queue_or_skip()
skip_if_dtype_not_supported(dt, q)
x = dpt.asarray([-1, 1], dtype=dpt.dtype(dt), sycl_queue=q)
r = dpt.cumulative_sum(x, dtype="?")
assert dpt.all(r)