tvm_tirx.testing.env.has_multi_gpu

Contents

tvm_tirx.testing.env.has_multi_gpu#

tvm_tirx.testing.env.has_multi_gpu(count=2)#

True if at least count devices of a single GPU backend exist.

Parameters:

count (int)

Return type:

bool