org.apache.tvm

Class TVMContext

java.lang.Object

org.apache.tvm.TVMContext

Direct Known Subclasses:

TVMRemoteContext

public class TVMContext
extends Object

Field Summary

Fields

Modifier and Type	Field and Description
int	deviceId
int	deviceType

Constructor Summary

Constructors

Constructor and Description
TVMContext(int deviceType, int deviceId)
TVMContext(String deviceType, int deviceId)

Method Summary

Modifier and Type	Method and Description
static TVMContext	cpu()
static TVMContext	cpu(int devId) Construct a CPU device.
boolean	equals(Object other)
boolean	exist() Whether this device exists.
static TVMContext	gpu()
static TVMContext	gpu(int devId) Construct a GPU device.
int	hashCode()
static TVMContext	hexagon()
static TVMContext	hexagon(int devId) Construct a Hexagon device.
long	maxThreadsPerBlock() Maximum number of threads on each block.
static TVMContext	metal()
static TVMContext	metal(int devId) Construct a metal device.
static TVMContext	opencl()
static TVMContext	opencl(int devId) Construct a OpenCL device.
void	sync() Synchronize until jobs finished at the context.
String	toString()
static TVMContext	vpi()
static TVMContext	vpi(int devId) Construct a VPI simulated device.
static TVMContext	vulkan()
static TVMContext	vulkan(int devId) Construct a Vulkan device.
long	warpSize() Number of threads that executes in concurrent.

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Field Detail

deviceType

public final int deviceType

deviceId

public final int deviceId

Constructor Detail

TVMContext

public TVMContext(int deviceType,
                  int deviceId)

TVMContext

public TVMContext(String deviceType,
                  int deviceId)

Method Detail

cpu

public static TVMContext cpu(int devId)

Construct a CPU device.

Parameters:

devId - The device id

Returns:

The created context

cpu

public static TVMContext cpu()

gpu

public static TVMContext gpu(int devId)

Construct a GPU device.

Parameters:

devId - The device id

Returns:

The created context

gpu

public static TVMContext gpu()

opencl

public static TVMContext opencl(int devId)

Construct a OpenCL device.

Parameters:

devId - The device id

Returns:

The created context

opencl

public static TVMContext opencl()

vulkan

public static TVMContext vulkan(int devId)

Construct a Vulkan device.

Parameters:

devId - The device id

Returns:

The created context

vulkan

public static TVMContext vulkan()

metal

public static TVMContext metal(int devId)

Construct a metal device.

Parameters:

devId - The device id

Returns:

The created context

metal

public static TVMContext metal()

vpi

public static TVMContext vpi(int devId)

Construct a VPI simulated device.

Parameters:

devId - The device id

Returns:

The created context

vpi

public static TVMContext vpi()

hexagon

public static TVMContext hexagon(int devId)

Construct a Hexagon device.

Parameters:

devId - The device id

Returns:

The created context

hexagon

public static TVMContext hexagon()

exist

public boolean exist()

Whether this device exists.

Returns:

true if exists.

maxThreadsPerBlock

public long maxThreadsPerBlock()

Maximum number of threads on each block.

Returns:

the maximum thread number.

warpSize

public long warpSize()

Number of threads that executes in concurrent.

Returns:

the thread number.

sync

public void sync()

Synchronize until jobs finished at the context.

hashCode

public int hashCode()

Overrides:

hashCode in class Object

equals

public boolean equals(Object other)

Overrides:

equals in class Object

toString

public String toString()

Overrides:

toString in class Object