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

Merge pull request #19 from LokiMetaSmith/fix-amd-triton-reinstall

Browse Source 
Fix AMD Triton runtime error by reinstalling pytorch-triton-rocm
This commit is contained in:
Lawrence R Kincheloe III 2025-11-22 23:22:57 -06:00 committed by GitHub
commit 23695f817d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
speedrun.sh
View File

@ -46,6 +46,9 @@ source .venv/bin/activate
# if the NVIDIA version of triton (e.g. 3.4.0) is accidentally installed.
if [ "$EXTRAS" == "amd" ]; then
uv pip uninstall -q triton || true
# Uninstalling triton may have deleted the shared 'triton' directory, breaking pytorch-triton-rocm.
# Reinstall pytorch-triton-rocm to ensure it's intact.
uv pip install --force-reinstall --index-url https://repo.amd.com/rocm/whl/gfx1151 pytorch-triton-rocm
fi
# -----------------------------------------------------------------------------