From 48e8ccb67d3d7dbe213bf05051f63a0f9232df13 Mon Sep 17 00:00:00 2001
From: Tianqi Chen <tqchen@users.noreply.github.com>
Date: Tue, 22 Aug 2017 16:28:27 -0700
Subject: [PATCH] [API] Expose AutoInlineInjective (#368)
---
src/api/api_schedule.cc | 6 ++++++
1 file changed, 6 insertions(+)
diff --git a/src/api/api_schedule.cc b/src/api/api_schedule.cc
index 91a069bde..0b5ef2515 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) { \
--
GitLab