org.apache.tvm

Class Module

java.lang.Object

org.apache.tvm.TVMValue

org.apache.tvm.Module

public class Module
extends TVMValue

Container of compiled functions of TVM.

Field Summary

Fields

Modifier and Type	Field and Description
long	handle

Fields inherited from class org.apache.tvm.TVMValue

typeCode

Method Summary

Modifier and Type	Method and Description
Module	asModule() Easy for user to get the instance from returned TVMValue.
static boolean	enabled(String target) Whether module runtime is enabled for target, e.g., The following code checks if cuda is enabled.
Function	entryFunc() Get the entry function.
protected void	finalize()
Function	getFunction(String name)
Function	getFunction(String name, boolean queryImports) Get function from the module.
void	importModule(Module module) Add module to the import list of current one.
static Module	load(String path)
static Module	load(String path, String fmt) Load module from file.
void	release() Release the Module.
String	typeKey() Get type key of the module.

Methods inherited from class org.apache.tvm.TVMValue

asBytes, asDouble, asFunction, asLong, asNDArray, asString

Methods inherited from class java.lang.Object

clone, equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

handle

public final long handle

Method Detail

finalize

protected void finalize()
                 throws Throwable

Overrides:

finalize in class Object

Throws:

Throwable

asModule

public Module asModule()

Easy for user to get the instance from returned TVMValue.

Overrides:

asModule in class TVMValue

Returns:

this

release

public void release()

Release the Module.

We highly recommend you to do this manually since the GC strategy is lazy.

Overrides:

release in class TVMValue

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.