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

Fix NaN errors observed with ARIMA in CUDA 11.2 builds #3730

Merged
merged 2 commits into from
Apr 8, 2021

Conversation

Nyrio
Copy link
Contributor

@Nyrio Nyrio commented Apr 8, 2021

Closes #3649

The error was in batched_kalman_loop_kernel: calculating Y = a M X + b Y when b = 0 can still result in NaN if Y is uninitialized. One possible fix would be to initialize to zeros the component that was uninitialized. The fix I chose is to remove the unnecessary read, so no uninitialized value is accessed and we save unnecessary operations.

@Nyrio Nyrio requested review from a team as code owners April 8, 2021 16:52
@github-actions github-actions bot added CUDA/C++ Cython / Python Cython or Python issue labels Apr 8, 2021
@Nyrio Nyrio added 3 - Ready for Review Ready for review by team bug Something isn't working CUDA / C++ CUDA issue non-breaking Non-breaking change and removed Cython / Python Cython or Python issue CUDA / C++ CUDA issue labels Apr 8, 2021
@github-actions github-actions bot added the Cython / Python Cython or Python issue label Apr 8, 2021
@dantegd
Copy link
Member

dantegd commented Apr 8, 2021

rerun tests

@dantegd
Copy link
Member

dantegd commented Apr 8, 2021

@gpucibot merge

@rapids-bot rapids-bot bot merged commit 1291502 into rapidsai:branch-0.19 Apr 8, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
3 - Ready for Review Ready for review by team bug Something isn't working Cython / Python Cython or Python issue non-breaking Non-breaking change
Projects
None yet
Development

Successfully merging this pull request may close these issues.

[BUG] ARIMA pytest fails on CUDA 11.2
2 participants