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