[ASK FOR HELP] Does TensorIR support ROCm devices

Questions
1

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

2

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

3

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
4

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

6

oh，tvm is so powerful,ok

7

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.