Can I choose ‘rocm’ as the target device when I use TensorIR?
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.