feat: add AMD GPU detection for ROCm backend in LocalAGI installation scripts
This commit is contained in:
@@ -43,6 +43,10 @@ resolve_backend() {
|
|||||||
backend="cu128"
|
backend="cu128"
|
||||||
elif [[ -e /dev/kfd ]]; then
|
elif [[ -e /dev/kfd ]]; then
|
||||||
backend="rocm7.2"
|
backend="rocm7.2"
|
||||||
|
elif lspci 2>/dev/null | grep -qiE 'AMD|Radeon'; then
|
||||||
|
backend="rocm7.2"
|
||||||
|
elif grep -qEi '0x1002|0x1022' /sys/class/drm/renderD*/device/vendor /sys/class/drm/card*/device/vendor 2>/dev/null; then
|
||||||
|
backend="rocm7.2"
|
||||||
fi
|
fi
|
||||||
fi
|
fi
|
||||||
;;
|
;;
|
||||||
|
|||||||
@@ -40,6 +40,10 @@ resolve_backend() {
|
|||||||
backend="cu128"
|
backend="cu128"
|
||||||
elif [[ -e /dev/kfd ]]; then
|
elif [[ -e /dev/kfd ]]; then
|
||||||
backend="rocm7.2"
|
backend="rocm7.2"
|
||||||
|
elif lspci 2>/dev/null | grep -qiE 'AMD|Radeon'; then
|
||||||
|
backend="rocm7.2"
|
||||||
|
elif grep -qEi '0x1002|0x1022' /sys/class/drm/renderD*/device/vendor /sys/class/drm/card*/device/vendor 2>/dev/null; then
|
||||||
|
backend="rocm7.2"
|
||||||
fi
|
fi
|
||||||
fi
|
fi
|
||||||
;;
|
;;
|
||||||
|
|||||||
Reference in New Issue
Block a user