-
Notifications
You must be signed in to change notification settings - Fork 621
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 failing test for Numba Func #2893
Conversation
Signed-off-by: Rafal Maj <rmaj@nvidia.com>
Signed-off-by: Rafal Maj <rmaj@nvidia.com>
Signed-off-by: Rafal Maj <rmaj@nvidia.com>
!build |
CI MESSAGE: [2294233]: BUILD STARTED |
CI MESSAGE: [2294233]: BUILD PASSED |
def test_multiple_ins(): | ||
pipe = numba_multiple_ins_pipe(shapes=(10, 10), dtype=np.uint8, batch_size=8, num_threads=1, device_id=0, run_fn=multiple_ins_run, setup_fn=multiple_ins_setup, | ||
pipe = numba_multiple_ins_pipe(shapes=[(10, 10)], dtype=np.uint8, batch_size=8, num_threads=1, device_id=0, run_fn=multiple_ins_run, setup_fn=multiple_ins_setup, |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Is this the fix?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Yes, that fixes test. I populated data in wrong way. That's why I added additional checks basing on what data type user pass and what user expect.
Fix for dead ptr is adding out_shapes_ member variable and holding pointers to it.
Signed-off-by: Rafal Maj <rmaj@nvidia.com>
!build |
CI MESSAGE: [2294485]: BUILD STARTED |
CI MESSAGE: [2294485]: BUILD PASSED |
Why we need this PR?
What happened in this PR?
[ Passed correct data to test and add checks if data passed by user is in same shape that user expects ]
[ Additional checks in numba_func.cc ]
[ NA ]
[ NA ]
[ NA ]