tvm_tirx.runtime.rocm

tvm_tirx.runtime.rocm(dev_id=0)

Construct a ROCM device

Parameters:

dev_id (int, optional) – The integer device id

Returns:

dev (Device) – The created device