True if a ROCm device is present and enabled in TVM_TEST_TARGETS.
()
| 178 | |
| 179 | |
| 180 | def has_rocm() -> bool: |
| 181 | """True if a ROCm device is present and enabled in TVM_TEST_TARGETS.""" |
| 182 | return _device_usable("rocm") |
| 183 | |
| 184 | |
| 185 | def has_vulkan() -> bool: |
no test coverage detected
searching dependent graphs…