-
Notifications
You must be signed in to change notification settings - Fork 3k
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 QNN EP HTP shared memory allocator #23136
Conversation
… declarations and definitions for IAllocator::TensorAlloc().
…ion clean up callback
onnxruntime/core/providers/qnn/builder/qnn_context_mem_handle_manager.cc
Show resolved
Hide resolved
onnxruntime/core/providers/qnn/builder/qnn_context_mem_handle_manager.cc
Show resolved
Hide resolved
Can this be used for Lora support when the model is modified to have optional inputs, and the data can be fed to override default initializers? |
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.
🕐
I'm not too familiar with the scenario. If that can be done using OrtValues, an input OrtValue can use this new allocator. |
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.
/azp run Windows GPU WebGPU CI Pipeline |
Azure Pipelines successfully started running 1 pipeline(s). |
Description
Adds QNN EP HTP shared memory allocator.
The HTP shared memory allocator (
HtpSharedMemoryAllocator
) calls the rpcmem shared library (libcdsprpc.so/dll) to allocate and free memory that can be shared between HTP and CPU.The allocator can be enabled by setting QNN EP option
enable_htp_shared_memory_allocator
to1
.QNNExecutionProvider::CreatePreferredAllocators()
will then return an instance ofHtpSharedMemoryAllocator
.For each QNN context, we also need to register and unregister memory handles in order to use the HTP shared memory. This memory handle management is added to
QnnBackendManager
, which also manages the QNN context handles.For more information about using HTP shared memory with QNN, see: https://docs.qualcomm.com/bundle/publicresource/topics/80-63442-50/htp_shared_buffer_tutorial.html#shared-buffer-tutorial
Limitations:
Motivation and Context
Improve performance by using HTP shared memory to avoid overhead from copying data between CPU and NPU.