immich/machine-learning/app
Mert 2b37caba03
feat(ml): rocm ()
* feat(ml): introduce support of onnxruntime-rocm for AMD GPU

* try mutex for algo cache

use OrtMutex

* bump versions, run on mich

use 3.12

use 1.19.2

* acquire lock before any changes can be made

guard algo benchmark results

mark mutex as mutable

re-add /bin/sh (?)

use 3.10

use 6.1.2

* use composite cache key

1.19.2

fix variable name

fix variable reference

aaaaaaaaaaaaaaaaaaaa

* bump deps

* disable algo caching

* fix gha

* try ubuntu runner

* actually fix the gha

* update patch

* skip mimalloc preload for rocm

* increase build threads

* increase timeout for rocm

* Revert "increase timeout for rocm"

This reverts commit 2c4452f5d132198ed381a7b262b4a5cab5114b5f.

* attempt migraphx

* set migraphx_home

* Revert "set migraphx_home"

This reverts commit c121d3e48754b3bce100636f8d666deec58a44b7.

* Revert "attempt migraphx"

This reverts commit 521f9fb72dbe506dc6cb8faeb6494817d87265c6.

* migraphx, take two

* bump rocm

* allow cpu

* try only targeting migraphx

* skip tests

* migraph 

* known issues

* target gfx900 and gfx1102

* mention `HSA_USE_SVM`

* update lock

* set device id for rocm

---------

Co-authored-by: Mehdi GHESH <mehdi.ghesh@hotmail.fr>
2025-03-17 21:08:19 +00:00
..
models feat(ml): rocm () 2025-03-17 21:08:19 +00:00
sessions feat(ml): rocm () 2025-03-17 21:08:19 +00:00
__init__.py refactor(ml): modularization and styling () 2023-06-24 22:18:09 -05:00
config.py feat(ml): ML on Rockchip NPUs () 2025-03-17 12:04:08 -04:00
conftest.py feat(ml): ML on Rockchip NPUs () 2025-03-17 12:04:08 -04:00
healthcheck.py feat: add docker healthchecks to server and ml () 2024-05-22 16:54:29 +00:00
main.py feat(ml): ML on Rockchip NPUs () 2025-03-17 12:04:08 -04:00
schemas.py feat(ml): ML on Rockchip NPUs () 2025-03-17 12:04:08 -04:00
test_main.py feat(ml): rocm () 2025-03-17 21:08:19 +00:00