administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-15 07:35:27 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

rocm updated to 6.2.1

Browse Source
This commit is contained in:
Artur Braga 2024-09-26 13:48:55 +02:00
parent fc3aac96f2
commit 54cdec07e8
5 changed files with 10 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
docker/rocm/Dockerfile
View File

@ -2,7 +2,7 @@
# https://askubuntu.com/questions/972516/debian-frontend-environment-variable
ARG DEBIAN_FRONTEND=noninteractive
ARG ROCM=5.7.3
ARG ROCM=6.2.1
ARG AMDGPU=gfx900
ARG HSA_OVERRIDE_GFX_VERSION
ARG HSA_OVERRIDE
@ -23,11 +23,12 @@ COPY docker/rocm/rocm-pin-600 /etc/apt/preferences.d/
RUN apt-get update
RUN apt-get -y install --no-install-recommends migraphx hipfft roctracer
RUN apt-get -y install --no-install-recommends migraphx hipfft roctracer rocprofiler-register miopen-hip
RUN apt-get -y install --no-install-recommends migraphx-dev
RUN mkdir -p /opt/rocm-dist/opt/rocm-$ROCM/lib
RUN cd /opt/rocm-$ROCM/lib && cp -dpr libMIOpen*.so* libamd*.so* libhip*.so* libhsa*.so* libmigraphx*.so* librocm*.so* librocblas*.so* libroctracer*.so* librocfft*.so* /opt/rocm-dist/opt/rocm-$ROCM/lib/
RUN mkdir -p /opt/rocm-dist/opt/rocm-$ROCM/lib/migraphx/lib
RUN cd /opt/rocm-$ROCM/lib && cp -dpr libMIOpen*.so* libamd*.so* libhip*.so* libhsa*.so* librocm*.so* librocblas*.so* libroctracer*.so* libroctx64*.so* librocfft*.so* librocprofiler-register*.so* /opt/rocm-dist/opt/rocm-$ROCM/lib/
RUN cd /opt/rocm-$ROCM/lib/migraphx/lib && cp -dpr libmigraphx*.so* /opt/rocm-dist/opt/rocm-$ROCM/lib/migraphx/lib/
RUN cd /opt/rocm-dist/opt/ && ln -s rocm-$ROCM rocm
RUN mkdir -p /opt/rocm-dist/etc/ld.so.conf.d/

4
docker/rocm/migraphx/CMakeLists.txt
View File

@ -13,12 +13,12 @@ SET(CMAKE_INSTALL_RPATH_USE_LINK_PATH TRUE)
project(migraphx_py)
include_directories(/opt/rocm/include)
include_directories(/opt/rocm/include /opt/rocm/lib/migraphx/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)
target_link_libraries(migraphx PRIVATE /opt/rocm/lib/migraphx/lib/libmigraphx.so.2010000 /opt/rocm/lib/migraphx/lib/libmigraphx_tf.so.2010000 /opt/rocm/lib/migraphx/lib/libmigraphx_onnx.so.2010000)
install(TARGETS migraphx
COMPONENT python

2
docker/rocm/migraphx/migraphx_py.cpp
View File

@ -554,7 +554,7 @@ MIGRAPHX_PYBIND11_MODULE(migraphx, m)
a.fill(values.begin(), values.end());
return a;
});
m.def("generate_argument", &migraphx::generate_argument, py::arg("s"), py::arg("seed") = 0);
m.def("generate_argument", &migraphx::generate_argument, py::arg("s"), py::arg("seed") = 0,py::arg("random_mode::random"));
m.def("fill_argument", &migraphx::fill_argument, py::arg("s"), py::arg("value"));
m.def("quantize_fp16",
&migraphx::quantize_fp16,

2
docker/rocm/rocm.hcl
View File

@ -2,7 +2,7 @@ variable "AMDGPU" {
default = "gfx900"
}
variable "ROCM" {
default = "5.7.3"
default = "6.2.1"
}
variable "HSA_OVERRIDE_GFX_VERSION" {
default = ""

2
docker/rocm/rocm.list
View File

@ -1 +1 @@
deb [arch=amd64 signed-by=/etc/apt/keyrings/rocm.gpg] https://repo.radeon.com/rocm/apt/5.7.3 focal main
deb [arch=amd64 signed-by=/etc/apt/keyrings/rocm.gpg] https://repo.radeon.com/rocm/apt/6.2.1 focal main