Patch lite custom op API (microsoft#17605)

A few enhancements: - Support compute returning status; - Support variadic; --------- Co-authored-by: Randy Shuai <[email protected]>
AmusementClub · Sep 26, 2023 · 37dcefb · 37dcefb
1 parent 4ab0e17
commit 37dcefb
Show file tree

Hide file tree

Showing 7 changed files with 656 additions and 211 deletions.
diff --git a/include/onnxruntime/core/session/onnxruntime_cxx_api.h b/include/onnxruntime/core/session/onnxruntime_cxx_api.h
@@ -2055,6 +2055,7 @@ struct KernelContext {
   void* GetGPUComputeStream() const;
   Logger GetLogger() const;
   OrtAllocator* GetAllocator(const OrtMemoryInfo& memory_info) const;
+  OrtKernelContext* GetOrtKernelContext() const { return ctx_; }
 
  private:
   OrtKernelContext* ctx_;