-
Notifications
You must be signed in to change notification settings - Fork 607
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
Changes to dali/util/numpy #5416
Conversation
- Make `numpy.h` available to users - Extend `ReadTensor` to also accept a `bool pinned` argument - See NVIDIA#5337 for details Signed-off-by: Francesco Versaci <francesco.versaci@gmail.com>
Hi @fversaci, Thank you for your contribution. Looks good. I have one small ask - can you sign off your commit? |
Hi, I should have fixed that (via a |
Sweet. Thanks! |
!build |
CI MESSAGE: [14025578]: BUILD STARTED |
CI MESSAGE: [14025578]: BUILD PASSED |
Stop using pinned-memory in CPU-only test Co-authored-by: Michał Zientkiewicz <mzient@gmail.com> Signed-off-by: Francesco Versaci <francesco.versaci@gmail.com>
!build |
CI MESSAGE: [14101833]: BUILD STARTED |
CI MESSAGE: [14101833]: BUILD FAILED |
CI MESSAGE: [14101833]: BUILD PASSED |
Category:
New feature (non-breaking change which adds functionality)
Description:
numpy.h
available to usersReadTensor
to also accept abool pinned
argumentAdditional 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