Faaz commited on
Commit
4a33f96
·
1 Parent(s): 161c946

Remove HSA_OVERRIDE_GFX_VERSION - ROCm 7.0 native MI300X support

Browse files
Files changed (1) hide show
  1. 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
- export HSA_OVERRIDE_GFX_VERSION=11.0.0
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