tvm_tirx.testing.env.cuda_arch

Contents

tvm_tirx.testing.env.cuda_arch#

tvm_tirx.testing.env.cuda_arch(device_id=0)#

Return a CUDA device’s canonical runtime architecture, if available.

Architecture-specific targets use the same spelling as CUDA compilation targets (for example, ‘sm_100a’ and ‘sm_103a’), rather than collapsing them to a major compute-capability floor.

Parameters:

device_id (int)

Return type:

str | None