Skip to content
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.

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

dask.array.tensordot bugfix: ensure axes are positive / add tests for negative axes #10812

Open
wants to merge 2 commits into
base: main
Choose a base branch
from

Conversation

joanrue
Copy link
Contributor

@joanrue joanrue commented Jan 17, 2024

While numpy.tensordot accepts negative axes, negative axes in dask.array.tensordot were giving wrong results. This bug is easily fixed by ensuring that the left axes are positive before the sum-reduction step.

Co-authored-by: Sepand Kashani sepand@kashani.ch

tensordot in numpy accepts negative axes, however negative axes in dask.array.tensordot these give wrong results.
@GPUtester
Copy link
Collaborator

Can one of the admins verify this patch?

Admins can comment ok to test to allow this one PR to run or add to allowlist to allow all future PRs from the same author to run.

Copy link
Contributor

github-actions bot commented Jan 17, 2024

Unit Test Results

See test report for an extended history of previous test failures. This is useful for diagnosing flaky tests.

     15 files  ±0       15 suites  ±0   3h 22m 59s ⏱️ +18s
 13 123 tests ±0   12 158 ✅ ±0     931 💤 ±0  34 ❌ ±0 
162 520 runs  ±0  142 449 ✅ ±0  20 037 💤 ±0  34 ❌ ±0 

For more details on these failures, see this check.

Results for commit 83e636f. ± Comparison against base commit a291fd2.

♻️ This comment has been updated with latest results.

@joanrue joanrue changed the title tensordot bugfix: ensure axes are positive / add tests for negative axes dask.array.tensordot bugfix: ensure axes are positive / add tests for negative axes Jan 19, 2024
@hendrikmakait hendrikmakait self-requested a review January 23, 2024 15:40
@joanrue
Copy link
Contributor Author

joanrue commented Mar 21, 2024

Hi! I'm reaching out to see if there's anything more I can do to facilitate the review process.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

dask.array.tensordot bug: negative values in argument axes give wrong results.
2 participants