[JVM] Update the runtime PackedFunc for module

apache · Feb 12, 2020 · 79cfab0 · 79cfab0
1 parent aaf62e4
commit 79cfab0
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/jvm/core/src/main/java/org/apache/tvm/Module.java b/jvm/core/src/main/java/org/apache/tvm/Module.java
@@ -38,7 +38,7 @@ protected Map<String, Function> initialValue() {
   private static Function getApi(String name) {
     Function func = apiFuncs.get().get(name);
     if (func == null) {
-      func = Function.getFunction("module." + name);
+      func = Function.getFunction("runtime." + name);
       apiFuncs.get().put(name, func);
     }
     return func;
@@ -126,7 +126,7 @@ public void importModule(Module module) {
    * @return type key of the module.
    */
   public String typeKey() {
-    return getApi("_GetTypeKey").pushArg(this).invoke().asString();
+    return getApi("ModuleGetTypeKey").pushArg(this).invoke().asString();
   }
 
   /**
@@ -137,7 +137,7 @@ public String typeKey() {
    * @return The loaded module
    */
   public static Module load(String path, String fmt) {
-    TVMValue ret = getApi("_LoadFromFile").pushArg(path).pushArg(fmt).invoke();
+    TVMValue ret = getApi("ModuleLoadFromFile").pushArg(path).pushArg(fmt).invoke();
     assert ret.typeCode == TypeCode.MODULE_HANDLE;
     return ret.asModule();
   }
@@ -154,7 +154,7 @@ public static Module load(String path) {
    * @return Whether runtime is enabled.
    */
   public static boolean enabled(String target) {
-    TVMValue ret = getApi("_Enabled").pushArg(target).invoke();
+    TVMValue ret = getApi("RuntimeEnabled").pushArg(target).invoke();
     return ret.asLong() != 0;
   }
 }