Skip to content

Update shared memory bound checking for infer requests - #565

Merged
jbkyang-nvi merged 9 commits into
mainfrom
kyang-validate-shm-byte-size
Apr 12, 2024
Merged

Update shared memory bound checking for infer requests#565
jbkyang-nvi merged 9 commits into
mainfrom
kyang-validate-shm-byte-size

Conversation

@jbkyang-nvi

Copy link
Copy Markdown
Contributor

@jbkyang-nvi
jbkyang-nvi requested a review from Tabrizian April 9, 2024 02:57
Comment thread src/c++/perf_analyzer/client_backend/triton_c_api/triton_loader.cc
@Tabrizian

Copy link
Copy Markdown
Member

Could you update the PR description to better reflect the changes?

@jbkyang-nvi
jbkyang-nvi force-pushed the kyang-validate-shm-byte-size branch from c19ea39 to 9f29920 Compare April 10, 2024 00:28
Comment thread src/c++/perf_analyzer/client_backend/triton_c_api/shared_memory_manager.h Outdated
Comment thread src/c++/perf_analyzer/client_backend/triton_c_api/shared_memory_manager.h Outdated
jbkyang-nvi and others added 4 commits April 10, 2024 10:24
Co-authored-by: Iman Tabrizian <iman.tabrizian@gmail.com>
Co-authored-by: Iman Tabrizian <iman.tabrizian@gmail.com>
…rence-server/client into kyang-validate-shm-byte-size
@jbkyang-nvi jbkyang-nvi changed the title Update according to server changes Update shared memory bound checking for infer requests Apr 10, 2024
jbkyang-nvi added a commit to triton-inference-server/server that referenced this pull request Apr 12, 2024
…nds (#7083)

Validate that memory offset and byte size requested is not out of bounds 
of registered memory. 
Previously in #6914 we checked out of bounds offset for shared memory
requests. This PR also adds more testing to verify the block of memory 
is in fact in bounds.
Client change: triton-inference-server/client#565
@jbkyang-nvi
jbkyang-nvi merged commit d8b4077 into main Apr 12, 2024
@jbkyang-nvi
jbkyang-nvi deleted the kyang-validate-shm-byte-size branch April 12, 2024 18:24
jbkyang-nvi added a commit to triton-inference-server/server that referenced this pull request Apr 12, 2024
…nds (#7083)

Validate that memory offset and byte size requested is not out of bounds 
of registered memory. 
Previously in #6914 we checked out of bounds offset for shared memory
requests. This PR also adds more testing to verify the block of memory 
is in fact in bounds.
Client change: triton-inference-server/client#565
mc-nv pushed a commit to triton-inference-server/server that referenced this pull request Apr 12, 2024
…nds (#7083) (#7111)

Validate that memory offset and byte size requested is not out of bounds 
of registered memory. 
Previously in #6914 we checked out of bounds offset for shared memory
requests. This PR also adds more testing to verify the block of memory 
is in fact in bounds.
Client change: triton-inference-server/client#565
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Development

Successfully merging this pull request may close these issues.

2 participants