zishuo/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-09-26 19:41:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update ROCm to not hang when running on complex RNN models (#20118)

Browse Source 
* Update ROCm to not hang when running on complex RNN models

* Formatting
This commit is contained in:
Nicolas Mowen
2025-09-17 18:26:32 -06:00
committed by GitHub
parent 1efff67e32
commit c05e260ae9
2 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
docker/rocm/Dockerfile
View File

@@ -67,6 +67,9 @@ FROM deps-prelim AS rocm-prelim-hsa-override0
ENV HSA_ENABLE_SDMA=0
ENV TF_ROCM_USE_IMMEDIATE_MODE=1
# avoid kernel crashes
ENV HIP_FORCE_DEV_KERNARG=1
COPY --from=rocm-dist / /
RUN ldconfig