[API] Expose AutoInlineInjective (#368)

48e8ccb6 · Tianqi Chen · GitHub · b3bb8126 · 48e8ccb6
Commit 48e8ccb6 authored Aug 22, 2017 by Tianqi Chen Committed by GitHub Aug 22, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 0 deletions

src/api/api_schedule.cc
+6 -0

No files found.
--- a/src/api/api_schedule.cc
+++ b/src/api/api_schedule.cc
@@ -18,6 +18,12 @@ TVM_REGISTER_API("schedule.AutoInlineElemWise")
    AutoInlineElemWise(args[0]);
  });

+
+TVM_REGISTER_API("schedule.AutoInlineInjective")
+.set_body([](TVMArgs args, TVMRetValue* ret) {
+    AutoInlineInjective(args[0]);
+  });
+
 #define REGISTER_SCHEDULE_PASS1(PassName)                         \
  TVM_REGISTER_API("schedule."#PassName)                          \
  .set_body([](TVMArgs args,  TVMRetValue *ret) {                 \