onnxruntime
[EP ABI] Add OrtKernelInfo APIs for kernel-based plugin EPs
#26803
Merged

[EP ABI] Add OrtKernelInfo APIs for kernel-based plugin EPs #26803

adrianlizarraga
adrianlizarraga [EP ABI] Add weight pre-packing support to kernel-based plugin EPs
4f3d0042
adrianlizarraga Add comment about sharing of prepacked weights (cpu ep only)
e9540216
adrianlizarraga Update Mul kernel to pre-pack input b
fb2998b6
adrianlizarraga Apply suggestions from code review
5e64f79c
adrianlizarraga Add comments regarding prepack allocator lifetime
9b1c6a20
adrianlizarraga Merge branch 'adrianl/plugin-ep-kernel-prepack' of github.com:microso…
c638a1a0
adrianlizarraga Added support for sharing pre-packed weights for cpu-accessible alloc…
717ed4ac
adrianlizarraga Define what should happen if OrtKernelImpl::SetSharedPrePackedWeight(…
bd8f6f04
adrianlizarraga Add some missing OrtKernelInfo APIs for plugin EPs
5c94ec47
adrianlizarraga
adrianlizarraga commented on 2025-12-15
adrianlizarraga Merge branch 'main' into adrianl/plugin-ep-kernel-prepack
fc1fd161
adrianlizarraga Clean up some exception handling
8b3f56c6
adrianlizarraga Refactor example kernel classes (no inheritance)
23503a18
adrianlizarraga Merge branch 'main' into adrianl/plugin-ep-kernel-prepack
7f37ffbd
adrianlizarraga Correct use of output param
26eca56c
adrianlizarraga Add more edge-case handling for PrePack() call
7af257b3
adrianlizarraga API version checks
515062e9
adrianlizarraga Use correct SAL annotation for array parameters
347ce4f4
adrianlizarraga Clean up some includes
906187d3
adrianlizarraga Merge branch 'main' into adrianl/KernelPluginEp_KernelInfoApis
56074f61
adrianlizarraga Update onnxruntime/core/session/plugin_ep/ep_kernel_registration.cc
1611fc38
adrianlizarraga Rename ExecuteIfCustomOpsApiEnabled to ExecuteifKernelApiEnabled
cb75d5a4
adrianlizarraga Remove OrtAllocator parameter from SharedPrePackedWeightCache_StoreWe…
30ca590e
adrianlizarraga Clarify what happens when SharedPrePackedWeightCache_StoreWeightData …
a5342b92
adrianlizarraga Merge branch 'main' into adrianl/KernelPluginEp_KernelInfoApis
c42d7ac6
adrianlizarraga Move some kernel info apis to OrtApi; Fix unrelated memory leak for K…
33233305
adrianlizarraga Merge branch 'main' into adrianl/plugin-ep-kernel-prepack
51bc7319
adrianlizarraga Review comments
edf3f2c2
adrianlizarraga C++ API
e94c0aab
adrianlizarraga Improve doc for c++ api convenience class
c8eb3c90
adrianlizarraga Add buffer_sizes as a parameter to OrtKernelImpl::SetSharedWeightData
98e3d132
adrianlizarraga Add comment to implementation of OrtKernelImpl::SetSharedPrePackedWeight
c61ae41f
adrianlizarraga Do not prescribe what the kernel impl should return for a situation t…
02d75d2e
adrianlizarraga Update include/onnxruntime/core/session/onnxruntime_ep_c_api.h
0a84eda6
adrianlizarraga Adjust comments
5f80f9d3
adrianlizarraga Tweak comment again
c60472d3
adrianlizarraga Merge main and fix conflicts
12b8394f
adrianlizarraga Update example kernel EP to get the OrtEp from OrtKernelInfo and add …
52453f2e
adrianlizarraga adrianlizarraga changed the base branch from main to adrianl/plugin-ep-kernel-prepack 218 days ago
adrianlizarraga adrianlizarraga requested a review from copilot-pull-request-reviewer copilot-pull-request-reviewer 218 days ago
copilot-pull-request-reviewer
copilot-pull-request-reviewer commented on 2025-12-26
adrianlizarraga Address copilot review comments
79eadd85
adrianlizarraga Address copilot review comment: update error message when APIs for Or…
1dc98b2b
adrianlizarraga Add comments to clarify ownership scenarios
441c9e25
adrianlizarraga Merge branch 'adrianl/plugin-ep-kernel-prepack' into adrianl/KernelPl…
5cf96000
Base automatically changed from adrianl/plugin-ep-kernel-prepack to main 217 days ago
fs-eire
adrianlizarraga Merge main and fix conflicts
ec0183e0
adrianlizarraga Add support for Add kernel to test kernel-based EP. This used as moti…
7e892619
adrianlizarraga Rename variable from 'mul_kernel' to 'binary_op_kernel'
1ddc9fb1
adrianlizarraga adrianlizarraga requested a review from copilot-pull-request-reviewer copilot-pull-request-reviewer 215 days ago
copilot-pull-request-reviewer
copilot-pull-request-reviewer commented on 2025-12-29
adrianlizarraga Address review comments and fix compiler warnings on Linux
9e6b06e4
adrianlizarraga adrianlizarraga requested a review from copilot-pull-request-reviewer copilot-pull-request-reviewer 215 days ago
copilot-pull-request-reviewer
copilot-pull-request-reviewer commented on 2025-12-29
adrianlizarraga Move KernelContext_GetAllocator memory leak fix to a different PR
1d496a29
adrianlizarraga Simplify binary op type check
6378bf26
adrianlizarraga adrianlizarraga marked this pull request as ready for review 214 days ago
adrianlizarraga adrianlizarraga requested a review from fs-eire fs-eire 214 days ago
adrianlizarraga adrianlizarraga requested a review from edgchen1 edgchen1 214 days ago
fs-eire
fs-eire dismissed these changes on 2026-01-04
edgchen1
edgchen1 commented on 2026-01-05
adrianlizarraga Merge branch 'main' into adrianl/KernelPluginEp_KernelInfoApis
eadc0290
adrianlizarraga Review comments + add new API to get op domain
cad57e7c
adrianlizarraga adrianlizarraga dismissed their stale review via cad57e7c 207 days ago
adrianlizarraga Check for NULL size arg in API call. Update comment
4fb2be94
adrianlizarraga adrianlizarraga requested a review from copilot-pull-request-reviewer copilot-pull-request-reviewer 207 days ago
copilot-pull-request-reviewer
copilot-pull-request-reviewer commented on 2026-01-06
adrianlizarraga Update SAL annotations for older KernelInfo_* API functions
ebe58ed6
adrianlizarraga Update example BinaryOp class to support Sub instead of Add
3968ec38
adrianlizarraga Update comment for KernelInfo_GetNodeName
df45a2f4
adrianlizarraga adrianlizarraga requested a review from copilot-pull-request-reviewer copilot-pull-request-reviewer 207 days ago
adrianlizarraga adrianlizarraga requested a review from edgchen1 edgchen1 207 days ago
copilot-pull-request-reviewer
copilot-pull-request-reviewer commented on 2026-01-06
edgchen1
edgchen1 approved these changes on 2026-01-06
adrianlizarraga adrianlizarraga enabled auto-merge (squash) 207 days ago
adrianlizarraga adrianlizarraga merged 27e6002d into main 207 days ago
adrianlizarraga adrianlizarraga deleted the adrianl/KernelPluginEp_KernelInfoApis branch 207 days ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone