-
Notifications
You must be signed in to change notification settings - Fork 629
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
Add pass-through tracking to auto-pinning buffers #4294
Conversation
Signed-off-by: Krzysztof Lecki <klecki@nvidia.com>
Signed-off-by: Krzysztof Lecki <klecki@nvidia.com>
!build |
CI MESSAGE: [6026863]: BUILD STARTED |
CI MESSAGE: [6026863]: BUILD FAILED |
CI MESSAGE: [6026863]: BUILD PASSED |
Signed-off-by: Krzysztof Lecki <klecki@nvidia.com>
be9f480
to
48d260b
Compare
!build |
CI MESSAGE: [6034109]: BUILD STARTED |
Signed-off-by: Krzysztof Lecki <klecki@nvidia.com>
!build |
CI MESSAGE: [6034120]: BUILD STARTED |
CI MESSAGE: [6034120]: BUILD PASSED |
!build |
CI MESSAGE: [6077156]: BUILD STARTED |
CI MESSAGE: [6077156]: BUILD FAILED |
!build |
CI MESSAGE: [6096195]: BUILD STARTED |
!build |
CI MESSAGE: [6096400]: BUILD STARTED |
CI MESSAGE: [6096400]: BUILD PASSED |
Category: New feature, Refactoring
Description:
Small refactor in executor - extract the backtracking of pass-through in the graph.
Originally it was used to obtain the group of tensor nodes that need to be treated
as outputs and buffered accordingly.
Use the functionality to pin the whole pass-through group when pinning a tensor.
Additional information:
Affected modules and functionalities:
Executor, graph
Key points relevant for the review:
Tests:
Make output buffers for arugment inputs to GPU operators pinned. #3728 didn't introduce additional tests for pinning, and I am not aware if we had any.
I'm not sure if we can easily observe the status of the outputs buffers outside of the operators or executor.
I might try to do something tomorrow, otherwise we will just have to observe the perf.I think I managed to write such test, inspecting the outputs of the executor for their pinnedness.
Checklist
Documentation
DALI team only
Requirements
REQ IDs: N/A
JIRA TASK: N/A