Faaz commited on
Commit ·
4a33f96
1
Parent(s): 161c946
Remove HSA_OVERRIDE_GFX_VERSION - ROCm 7.0 native MI300X support
Browse files- setup_mi300x.sh +1 -3
setup_mi300x.sh
CHANGED
|
@@ -95,7 +95,7 @@ echo ""
|
|
| 95 |
echo "[5/7] Setting environment variables ..."
|
| 96 |
|
| 97 |
# ROCm / PyTorch settings for MI300X
|
| 98 |
-
|
| 99 |
export PYTORCH_ROCM_ARCH="gfx942"
|
| 100 |
export HIP_VISIBLE_DEVICES=0
|
| 101 |
export TOKENIZERS_PARALLELISM=false
|
|
@@ -104,7 +104,6 @@ export WANDB_PROJECT="mindi-1.5-vision-coder"
|
|
| 104 |
# Create .env file for the project
|
| 105 |
cat > .env << EOF
|
| 106 |
HF_TOKEN=${HF_TOKEN}
|
| 107 |
-
HSA_OVERRIDE_GFX_VERSION=11.0.0
|
| 108 |
PYTORCH_ROCM_ARCH=gfx942
|
| 109 |
HIP_VISIBLE_DEVICES=0
|
| 110 |
TOKENIZERS_PARALLELISM=false
|
|
@@ -115,7 +114,6 @@ EOF
|
|
| 115 |
grep -q "HSA_OVERRIDE_GFX_VERSION" ~/.bashrc 2>/dev/null || cat >> ~/.bashrc << 'ENVEOF'
|
| 116 |
|
| 117 |
# MINDI 1.5 MI300X environment
|
| 118 |
-
export HSA_OVERRIDE_GFX_VERSION=11.0.0
|
| 119 |
export PYTORCH_ROCM_ARCH=gfx942
|
| 120 |
export HIP_VISIBLE_DEVICES=0
|
| 121 |
export TOKENIZERS_PARALLELISM=false
|
|
|
|
| 95 |
echo "[5/7] Setting environment variables ..."
|
| 96 |
|
| 97 |
# ROCm / PyTorch settings for MI300X
|
| 98 |
+
# NOTE: Do NOT set HSA_OVERRIDE_GFX_VERSION — ROCm 7.0 has native gfx942 support
|
| 99 |
export PYTORCH_ROCM_ARCH="gfx942"
|
| 100 |
export HIP_VISIBLE_DEVICES=0
|
| 101 |
export TOKENIZERS_PARALLELISM=false
|
|
|
|
| 104 |
# Create .env file for the project
|
| 105 |
cat > .env << EOF
|
| 106 |
HF_TOKEN=${HF_TOKEN}
|
|
|
|
| 107 |
PYTORCH_ROCM_ARCH=gfx942
|
| 108 |
HIP_VISIBLE_DEVICES=0
|
| 109 |
TOKENIZERS_PARALLELISM=false
|
|
|
|
| 114 |
grep -q "HSA_OVERRIDE_GFX_VERSION" ~/.bashrc 2>/dev/null || cat >> ~/.bashrc << 'ENVEOF'
|
| 115 |
|
| 116 |
# MINDI 1.5 MI300X environment
|
|
|
|
| 117 |
export PYTORCH_ROCM_ARCH=gfx942
|
| 118 |
export HIP_VISIBLE_DEVICES=0
|
| 119 |
export TOKENIZERS_PARALLELISM=false
|