diff --git a/src/api/api_schedule.cc b/src/api/api_schedule.cc
index 91a069bde8edf590b48054dd46a09b8adc062be4..0b5ef251503cf85e3b00f28e3c8bae3c6a14ad21 100644
--- 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) {                 \