[ASK FOR HELP] Does TensorIR support ROCm devices

Can I choose ‘rocm’ as the target device when I use TensorIR?

Yep. @masahi @vinx13 has lots of experience of it

There are also example using rocm tensor intrinsics here https://github.com/apache/tvm/blob/main/tests/python/integration/test_auto_tensorize.py#L308-L310

1 Like

Ok, thank you guys. I’m going to use HaiGuang DCU which seems to support ROCm, so I ask this question

oh,tvm is so powerful,ok

Hi, It is ok. https://tvm.apache.org/docs/reference/api/python/runtime.html?highlight=device#tvm.runtime.rocm Here we can use Rocm device. Thank you.