Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Hackathon 181] Add TVM support for FastDeploy #1811

Closed
wants to merge 8 commits into from
Closed

[Hackathon 181] Add TVM support for FastDeploy #1811

wants to merge 8 commits into from

Conversation

Zheng-Bicheng
Copy link
Collaborator

PR types(PR类型)

Backend

Description

@Zheng-Bicheng
Copy link
Collaborator Author

由于PPYOLOE比较大,编译时间较长,本次测试模型为Picodet,最后的测试会用PPYOLOE来做,测试结果如下图所示:

infer

@DefTruth DefTruth changed the title [Hackathon 181]Add tvm [Hackathon 181] Add TVM support for FastDeploy Apr 13, 2023
CMakeLists.txt Show resolved Hide resolved
@@ -167,6 +168,10 @@ if(ENABLE_RKNPU2_BACKEND)
list(APPEND FASTDEPLOY_LIBS ${RKNPU2_LIB})
endif()

if(ENABLE_TVM_BACKEND)
set(TVM_RUNTIME_LIB ${CMAKE_CURRENT_LIST_DIR}/third_libs/install/tvm/lib/libtvm_runtime.dylib)
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

dylib是Mac的动态库,这里是只支持Mac?Linux和Win怎么支持呢?这里需要按照OS来做下处理。

Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

这个需要编译下,我想想咋解决

@@ -0,0 +1,17 @@
# set path
set(TVM_BASE_PATH "/Users/zhengbicheng/Code/Hackathon/AddTVMForFastDeploy/FastDeploy/tvm")
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

这里不能hardcode,参考paddlelite.cmake来写吧,从某个固定的地址下载。

docs/cn/build_and_install/tvm.md Show resolved Hide resolved
docs/cn/build_and_install/tvm.md Outdated Show resolved Hide resolved
fastdeploy/runtime/backends/tvm/tvm_backend.cc Outdated Show resolved Hide resolved
fastdeploy/runtime/backends/tvm/tvm_backend.cc Outdated Show resolved Hide resolved
fastdeploy/runtime/backends/tvm/tvm_backend.cc Outdated Show resolved Hide resolved
fastdeploy/runtime/backends/tvm/tvm_backend.cc Outdated Show resolved Hide resolved
fastdeploy/runtime/runtime.cc Show resolved Hide resolved
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants