Package org.apache.tvm.rpc

Class TVMRemoteDevice

java.lang.Object

org.apache.tvm.TVMValue

org.apache.tvm.TVMObject

org.apache.tvm.rpc.TVMRemoteDevice

public class TVMRemoteDevice
extends TVMObject

A session-owned remote device, distinct from a local Device. Its native object retains the session and an ordinary endpoint device type and index. The endpoint descriptor has no session identity by itself. Allocation, queries and synchronization use the owning session and fail after its closure.

Field Summary

Fields inherited from class org.apache.tvm.TVMObject

handle, typeIndex

Constructor Summary

Constructors

Constructor	Description
TVMRemoteDevice(long handle, int typeIndex)

Method Summary

Modifier and Type	Method	Description
RemoteTensor	empty(long[] dimensions, TVMType dtype)	Allocate remote storage; shape metadata is constructed locally.
boolean	exist()
long	maxThreadsPerBlock()
void	sync()
long	warpSize()

Methods inherited from class org.apache.tvm.TVMObject

finalize, release

Methods inherited from class org.apache.tvm.TVMValue

asBytes, asDouble, asFunction, asLong, asModule, asString, asTensor

Methods inherited from class java.lang.Object

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

Constructor Details

TVMRemoteDevice

public TVMRemoteDevice(long handle,
 int typeIndex)

Method Details

exist

public boolean exist()

maxThreadsPerBlock

public long maxThreadsPerBlock()

warpSize

public long warpSize()

sync

public void sync()

empty

public RemoteTensor empty(long[] dimensions,
 TVMType dtype)

Allocate remote storage; shape metadata is constructed locally.