[PATCH] D115306: [NFC][mlgo] Generalize model runner interface

Yundi Qian via Phabricator via llvm-commits llvm-commits at lists.llvm.org
Wed Dec 8 16:07:34 PST 2021


yundiqian accepted this revision.
yundiqian added inline comments.
This revision is now accepted and ready to land.


================
Comment at: llvm/include/llvm/Analysis/MLModelRunner.h:41-45
+  virtual void *evaluateUntyped() = 0;
+  virtual void *getTensorUntyped(size_t Index) = 0;
+  const void *getTensorUntyped(size_t Index) const {
+    return (const_cast<MLModelRunner *>(this))->getTensorUntyped(Index);
+  }
----------------
do they need to be public?


================
Comment at: llvm/lib/Analysis/DevelopmentModeInlineAdvisor.cpp:269
 /// sacrificed for ease of use while training.
 class ModelUnderTrainingRunner final : public MLModelRunner {
 public:
----------------
will part of it get re-factored out of inline to common lib?


Repository:
  rG LLVM Github Monorepo

CHANGES SINCE LAST ACTION
  https://reviews.llvm.org/D115306/new/

https://reviews.llvm.org/D115306



More information about the llvm-commits mailing list