-
Notifications
You must be signed in to change notification settings - Fork 644
Add support for bool type for the numba operator #5666
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
Conversation
!build |
CI MESSAGE: [19138731]: BUILD STARTED |
CI MESSAGE: [19138731]: BUILD FAILED |
- adds support for bool types inside the numba operator - adds a test for that for CPU and GPU Signed-off-by: Janusz Lisiecki <[email protected]>
8d72fef
to
99f2802
Compare
!build |
CI MESSAGE: [19139148]: BUILD STARTED |
CI MESSAGE: [19139148]: BUILD PASSED |
for z in range(tz, out0.shape[0], z_s): | ||
for y in range(ty, out0.shape[1], y_s): | ||
for x in range(tx, out0.shape[2], x_s): | ||
out0[z][y][x] = 1 |
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 it how it's done in numba, or can/should we use a more pythonic/numpyic way?
for z in range(tz, out0.shape[0], z_s): | |
for y in range(ty, out0.shape[1], y_s): | |
for x in range(tx, out0.shape[2], x_s): | |
out0[z][y][x] = 1 | |
out0[tz::z_s, ty::y_s, tx::x_s] = 1 |
Signed-off-by: Janusz Lisiecki <[email protected]>
c357ad7
to
7242078
Compare
!build |
CI MESSAGE: [19173232]: BUILD STARTED |
CI MESSAGE: [19173232]: BUILD PASSED |
Category:
New feature (non-breaking change which adds functionality)
Description:
Additional information:
Affected modules and functionalities:
Key points relevant for the review:
Tests:
Checklist
Documentation
DALI team only
Requirements
REQ IDs: N/A
JIRA TASK: N/A