tvm_tirx.runtime.cuda

Contents

tvm_tirx.runtime.cuda#

tvm_tirx.runtime.cuda(dev_id=0)#

Construct a CUDA GPU device

Parameters:

dev_id (int, optional) – The integer device id

Returns:

dev (Device) – The created device