update

README.md

@@ -63,7 +63,12 @@ COCO2017（在网络良好的情况下，如果没有下载数据集，程序会
 
 CPP Deploy测试参考：
 ```
-    prepare_test_libs.py
+    cd yolov5s_tvm
+    mkdir lib
+    ulimit -s unlimited
+    export MIOPEN_DEBUG_CONV_IMPLICIT_GEMM=0
+    export HIP_VISIBLE_DEVICES=2(此处修改为对应加速卡ID号)
+    python3 prepare_test_libs.py
     bash run_example.sh
 ```
 Python Deploy测试参考：

run_example.sh

@@ -16,12 +16,11 @@
 # specific language governing permissions and limitations
 # under the License.
 echo "Build the libraries.."
-mkdir -p lib
 make
 echo "Run the example"
 export LD_LIBRARY_PATH=../../build:${LD_LIBRARY_PATH}
 export DYLD_LIBRARY_PATH=../../build:${DYLD_LIBRARY_PATH}
-export ROCBLAS_TENSILE_LIBPATH=/opt/dtk-22.10/lib/rocblas/library/
+export ROCBLAS_TENSILE_LIBPATH=/opt/dtk/lib/rocblas/library/
 
 
 

tvm_runtime_pack.cc

@@ -51,6 +51,8 @@
 #include "../../src/runtime/thread_pool.cc"
 #include "../../src/runtime/threading_backend.cc"
 #include "../../src/runtime/workspace_pool.cc"
+#include "../../src/runtime/contrib/miopen/conv_forward.cc"
+#include "../../src/runtime/contrib/miopen/miopen_utils.cc"
 #include "../../src/runtime/rocm/rocm_module.cc"
 #include "../../src/runtime/rocm/rocm_device_api.cc"