tacit/nanochat
1
0
Fork 0
mirror of https://github.com/karpathy/nanochat.git synced 2025-12-10 06:12:18 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update tinyrun.sh with AMD Strix Halo optimizations

Browse Source
This commit is contained in:
google-labs-jules[bot] 2025-11-23 22:08:02 +00:00
parent 146ce1a8f6
commit 6838699bd4
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
tinyrun.sh
View File

@ -61,6 +61,19 @@ if [ "$EXTRAS" == "amd" ]; then
export TRITON_HIP_LLD_PATH=$ROCM_LLD_PATH export TRITON_HIP_LLD_PATH=$ROCM_LLD_PATH
echo "Exported TRITON_HIP_LLD_PATH=$TRITON_HIP_LLD_PATH" echo "Exported TRITON_HIP_LLD_PATH=$TRITON_HIP_LLD_PATH"
fi fi
# AMD Strix Halo / APU specific settings
# Many APUs need this override to work with ROCm if the reported GFX version
# doesn't match the installed kernels exactly. Strix Halo is gfx1151.
# If users face issues, they might need to tweak this or use 11.0.0.
if [ -z "$HSA_OVERRIDE_GFX_VERSION" ]; then
export HSA_OVERRIDE_GFX_VERSION=11.5.1
echo "Exported HSA_OVERRIDE_GFX_VERSION=$HSA_OVERRIDE_GFX_VERSION (Strix Halo/APU compat)"
fi
# Disable SDMA to prevent system hangs on some APUs (common fix for Ryzen AI)
export HSA_ENABLE_SDMA=0
echo "Exported HSA_ENABLE_SDMA=0 (APU stability)"
fi fi
# ----------------------------------------------------------------------------- # -----------------------------------------------------------------------------