mirror of
https://github.com/BillyOutlast/rocm-automated.git
synced 2026-02-04 03:51:19 +01:00
debug build
This commit is contained in:
@@ -8,9 +8,9 @@ cd comfyui-manager && pip install -r requirements.txt && cd ..
|
|||||||
git clone -b AMD https://github.com/crystian/ComfyUI-Crystools.git
|
git clone -b AMD https://github.com/crystian/ComfyUI-Crystools.git
|
||||||
cd ComfyUI-Crystools && pip install -r requirements.txt && cd ..
|
cd ComfyUI-Crystools && pip install -r requirements.txt && cd ..
|
||||||
|
|
||||||
# MIGraphX
|
# MIGraphX DISABLED DUE TO CRASHES
|
||||||
git clone https://github.com/pnikolic-amd/ComfyUI_MIGraphX.git
|
#git clone https://github.com/pnikolic-amd/ComfyUI_MIGraphX.git
|
||||||
cd ComfyUI_MIGraphX && pip install -r requirements.txt && cd ..
|
#cd ComfyUI_MIGraphX && pip install -r requirements.txt && cd ..
|
||||||
|
|
||||||
# Unsafe Torch
|
# Unsafe Torch
|
||||||
git clone https://github.com/ltdrdata/comfyui-unsafe-torch
|
git clone https://github.com/ltdrdata/comfyui-unsafe-torch
|
||||||
|
|||||||
@@ -107,33 +107,33 @@ services:
|
|||||||
- HIP_VISIBLE_DEVICES=0
|
- HIP_VISIBLE_DEVICES=0
|
||||||
- RROCR_VISIBLE_DEVICES=1
|
- RROCR_VISIBLE_DEVICES=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
|
||||||
# === Memory allocator tuning ===
|
# === Memory allocator tuning ===
|
||||||
- PYTORCH_ALLOC_CONF=garbage_collection_threshold:0.6,max_split_size_mb:6144
|
# - PYTORCH_ALLOC_CONF=garbage_collection_threshold:0.6,max_split_size_mb:6144
|
||||||
# === Precision and performance ===
|
# === Precision and performance ===
|
||||||
- TORCH_BLAS_PREFER_HIPBLASLT=0
|
# - TORCH_BLAS_PREFER_HIPBLASLT=0
|
||||||
- TORCHINDUCTOR_MAX_AUTOTUNE_GEMM_BACKENDS="CK,TRITON,ROCBLAS"
|
# - TORCHINDUCTOR_MAX_AUTOTUNE_GEMM_BACKENDS="CK,TRITON,ROCBLAS"
|
||||||
- TORCHINDUCTOR_MAX_AUTOTUNE_GEMM_SEARCH_SPACE="BEST"
|
# - TORCHINDUCTOR_MAX_AUTOTUNE_GEMM_SEARCH_SPACE="BEST"
|
||||||
- TORCHINDUCTOR_FORCE_FALLBACK=0
|
# - TORCHINDUCTOR_FORCE_FALLBACK=0
|
||||||
# === Flash Attention ===
|
# === Flash Attention ===
|
||||||
- FLASH_ATTENTION_TRITON_AMD_ENABLE="TRUE"
|
# - FLASH_ATTENTION_TRITON_AMD_ENABLE="TRUE"
|
||||||
- FLASH_ATTENTION_BACKEND="flash_attn_triton_amd"
|
# - FLASH_ATTENTION_BACKEND="flash_attn_triton_amd"
|
||||||
- FLASH_ATTENTION_TRITON_AMD_SEQ_LEN=4096
|
# - FLASH_ATTENTION_TRITON_AMD_SEQ_LEN=4096
|
||||||
- USE_CK=ON
|
# - USE_CK=ON
|
||||||
- TRANSFORMERS_USE_FLASH_ATTENTION=1
|
# - TRANSFORMERS_USE_FLASH_ATTENTION=1
|
||||||
- TRITON_USE_ROCM=ON
|
# - TRITON_USE_ROCM=ON
|
||||||
- TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL=1
|
# - TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL=1
|
||||||
# === CPU threading ===
|
# === CPU threading ===
|
||||||
- OMP_NUM_THREADS=8
|
# - OMP_NUM_THREADS=8
|
||||||
- MKL_NUM_THREADS=8
|
# - MKL_NUM_THREADS=8
|
||||||
- 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=1
|
||||||
- MIOPEN_FIND_MODE=2
|
# - 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:
|
||||||
- "8188:8188"
|
- "8188:8188"
|
||||||
networks:
|
networks:
|
||||||
|
|||||||
Reference in New Issue
Block a user