Package org.apache.tvm

Class Module

java.lang.Object
org.apache.tvm.TVMValue
org.apache.tvm.TVMObject
org.apache.tvm.Module
public class Module extends TVMObject
Container of compiled functions of TVM.

  • Method Details

    • asModule

      public Module asModule()
      Easy for user to get the instance from returned TVMValue.
      Overrides:
      asModule in class TVMValue
      Returns:
      this

    • entryFunc

      public Function entryFunc()
      Get the entry function.
      Returns:
      The entry function if exist

    • getFunction

      public Function getFunction(String name, boolean queryImports)
      Get function from the module.
      Parameters:
      name - The name of the function.
      queryImports - Whether also query modules imported by this module.
      Returns:
      The result function.

    • getFunction

      public Function getFunction(String name)

    • importModule

      public void importModule(Module module)
      Add module to the import list of current one.
      Parameters:
      module - The other module.

    • typeKey

      public String typeKey()
      Get type key of the module.
      Returns:
      type key of the module.

    • load

      public static Module load(String path, String fmt)
      Load module from file.
      Parameters:
      path - The path to the module file.
      fmt - The format of the file, if not specified it will be inferred from suffix of the file.
      Returns:
      The loaded module

    • load

      public static Module load(String path)

    • enabled

      public static boolean enabled(String target)
      Whether module runtime is enabled for target, e.g., The following code checks if cuda is enabled. Module.enabled("cuda")
      Parameters:
      target - The target device type.
      Returns:
      Whether runtime is enabled.