chris/frigate
1
0
Fork 0
forked from Github/frigate
Code Pull Requests Packages Activity

ROCm AMD/GPU based build and detector, WIP

Browse Source
This commit is contained in:
Indrek Mandre
2024-01-25 20:14:22 +02:00
parent 487c626e00
commit 42f1168898
8 changed files with 845 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
docker/rocm/migraphx/CMakeLists.txt Normal file
View File

@@ -0,0 +1,26 @@
cmake_minimum_required(VERSION 3.1)
set(CMAKE_CXX_STANDARD 17)
set(CMAKE_CXX_STANDARD_REQUIRED ON)
set(CMAKE_CXX_EXTENSIONS OFF)
if(NOT CMAKE_BUILD_TYPE)
set(CMAKE_BUILD_TYPE Release)
endif()
SET(CMAKE_INSTALL_RPATH_USE_LINK_PATH TRUE)
project(migraphx_py)
include_directories(/opt/rocm/include)
find_package(pybind11 REQUIRED)
pybind11_add_module(migraphx migraphx_py.cpp)
target_link_libraries(migraphx PRIVATE /opt/rocm/lib/libmigraphx.so /opt/rocm/lib/libmigraphx_tf.so /opt/rocm/lib/libmigraphx_onnx.so)
install(TARGETS migraphx
COMPONENT python
LIBRARY DESTINATION /opt/rocm/lib
)