Skip to content

T
tic
Commit 48e8ccb6 by Tianqi Chen Committed by GitHub
Options

[API] Expose AutoInlineInjective (#368)

parent b3bb8126
Showing with 6 additions and 0 deletions
src/api/api_schedule.cc
...@@ -18,6 +18,12 @@ TVM_REGISTER_API("schedule.AutoInlineElemWise") ...@@ -18,6 +18,12 @@ TVM_REGISTER_API("schedule.AutoInlineElemWise")
AutoInlineElemWise(args[0]); AutoInlineElemWise(args[0]);
}); });
TVM_REGISTER_API("schedule.AutoInlineInjective")
.set_body([](TVMArgs args, TVMRetValue* ret) {
AutoInlineInjective(args[0]);
});
#define REGISTER_SCHEDULE_PASS1(PassName) \ #define REGISTER_SCHEDULE_PASS1(PassName) \
TVM_REGISTER_API("schedule."#PassName) \ TVM_REGISTER_API("schedule."#PassName) \
.set_body([](TVMArgs args, TVMRetValue *ret) { \ .set_body([](TVMArgs args, TVMRetValue *ret) { \
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment