mirror of
https://github.com/BillyOutlast/rocm-automated.git
synced 2026-02-04 03:51:19 +01:00
FIxing MIOPEN_FIND_MODE settings that can cause GPU hangs
This commit is contained in:
@@ -97,14 +97,16 @@ services:
|
|||||||
image: docker.io/getterup/comfyui-rocm7.1:latest
|
image: docker.io/getterup/comfyui-rocm7.1:latest
|
||||||
container_name: comfyui
|
container_name: comfyui
|
||||||
environment:
|
environment:
|
||||||
#- ROCR_VISIBLE_DEVICES=1
|
- HIP_VISIBLE_DEVICES=0
|
||||||
|
- ROCR_VISIBLE_DEVICES=0
|
||||||
- COMFYUI_ENABLE_ROCM=True
|
- COMFYUI_ENABLE_ROCM=True
|
||||||
- GPU_ARCH=rocm7.1
|
- GPU_ARCH=rocm7.1
|
||||||
- PYTORCH_TUNABLEOP_ENABLED=1
|
- PYTORCH_TUNABLEOP_ENABLED=0
|
||||||
- MIOPEN_FIND_MODE=NORMAL
|
- MIOPEN_FIND_MODE=1
|
||||||
- AMD_SERIALIZE_KERNEL=1
|
- AMD_SERIALIZE_KERNEL=1
|
||||||
- MIOPEN_USER_DB_PATH=/tmp/.miopen
|
- MIOPEN_USER_DB_PATH=/tmp/.miopen
|
||||||
- MIOPEN_CUSTOM_CACHE_DIR=/tmp/.miopen
|
- MIOPEN_CUSTOM_CACHE_DIR=/tmp/.miopen
|
||||||
|
- MIOPEN_DEBUG_DISABLE_FIND_DB=1
|
||||||
# === GPU targeting ===
|
# === GPU targeting ===
|
||||||
# - HCC_AMDGPU_TARGET="gfx1100" # Change for your GPU
|
# - HCC_AMDGPU_TARGET="gfx1100" # Change for your GPU
|
||||||
# - PYTORCH_ROCM_ARCH="gfx1100" # e.g., gfx1030 for RX 6800/6900
|
# - PYTORCH_ROCM_ARCH="gfx1100" # e.g., gfx1030 for RX 6800/6900
|
||||||
@@ -129,8 +131,7 @@ services:
|
|||||||
- NUMEXPR_NUM_THREADS=8
|
- NUMEXPR_NUM_THREADS=8
|
||||||
# === Experimental ROCm flags ===
|
# === Experimental ROCm flags ===
|
||||||
- HSA_ENABLE_ASYNC_COPY=1
|
- HSA_ENABLE_ASYNC_COPY=1
|
||||||
- HSA_ENABLE_SDMA=1
|
- HSA_ENABLE_SDMA=0
|
||||||
- MIOPEN_FIND_MODE=2
|
|
||||||
- MIOPEN_ENABLE_CACHE=1
|
- MIOPEN_ENABLE_CACHE=1
|
||||||
# - MIGRAPHX_MLIR_USE_SPECIFIC_OPS="attention"
|
# - MIGRAPHX_MLIR_USE_SPECIFIC_OPS="attention"
|
||||||
ports:
|
ports:
|
||||||
|
|||||||
Reference in New Issue
Block a user