Closed YongtaoHuang1994 closed 3 years ago
Please open a thread on discuss. The warning msg Warning: Unable to detect ROCm compute arch
suggests that there is some issue with rocm runtime. Make sure you have installed rocm correctly and that a normal rocm app works.
We have prepared the AMD GPU and its SDK ROCm.
Then we turn on "set(USE_LLVM llvm-config-9)" and "set(USE_ROCM on)" and build TVM with ROCm successfully. Next, we set the env variable:
After all the dependence is ready, we test resnet in TVM with ROCm. The inference code is shown as followed:
Then we get these errors:
Could you help me solve this problem? Thanks a lot.