tvm_tirx.testing.env.has_cuda_arch

Contents

tvm_tirx.testing.env.has_cuda_arch#

tvm_tirx.testing.env.has_cuda_arch(*arches, device_id=0)#

True if a CUDA device exactly matches one of arches.

Parameters:
  • arches (str)

  • device_id (int)

Return type:

bool