Commit b5ac9bba authored by Lei Wang's avatar Lei Wang Committed by LeiWang1999
Browse files

[Refactor] Phaseout redundant CUDA_DEVICE_ORDER export (#639)

parent e2d25ba8
...@@ -144,7 +144,6 @@ TILELANG_PATH="$(pwd)" ...@@ -144,7 +144,6 @@ TILELANG_PATH="$(pwd)"
echo "TileLang path set to: $TILELANG_PATH" echo "TileLang path set to: $TILELANG_PATH"
echo "Configuring environment variables for TVM..." echo "Configuring environment variables for TVM..."
echo "export PYTHONPATH=${TILELANG_PATH}:\$PYTHONPATH" >> ~/.bashrc echo "export PYTHONPATH=${TILELANG_PATH}:\$PYTHONPATH" >> ~/.bashrc
echo "export CUDA_DEVICE_ORDER=PCI_BUS_ID" >> ~/.bashrc
# Step 12: Source .bashrc to apply changes # Step 12: Source .bashrc to apply changes
echo "Applying environment changes by sourcing .bashrc..." echo "Applying environment changes by sourcing .bashrc..."
......
...@@ -91,10 +91,8 @@ cd .. ...@@ -91,10 +91,8 @@ cd ..
TILELANG_PATH="$(pwd)" TILELANG_PATH="$(pwd)"
echo "Configuring environment variables for TVM..." echo "Configuring environment variables for TVM..."
echo "export PYTHONPATH=${TILELANG_PATH}:\$PYTHONPATH" >> ~/.bashrc echo "export PYTHONPATH=${TILELANG_PATH}:\$PYTHONPATH" >> ~/.bashrc
echo "export CUDA_DEVICE_ORDER=PCI_BUS_ID" >> ~/.bashrc
TVM_HOME_ENV="export TVM_HOME=${TILELANG_PATH}/3rdparty/tvm" TVM_HOME_ENV="export TVM_HOME=${TILELANG_PATH}/3rdparty/tvm"
TILELANG_PYPATH_ENV="export PYTHONPATH=\$TVM_HOME/python:${TILELANG_PATH}:\$PYTHONPATH" TILELANG_PYPATH_ENV="export PYTHONPATH=\$TVM_HOME/python:${TILELANG_PATH}:\$PYTHONPATH"
CUDA_DEVICE_ORDER_ENV="export CUDA_DEVICE_ORDER=PCI_BUS_ID"
# Check and add the first line if not already present # Check and add the first line if not already present
if ! grep -qxF "$TVM_HOME_ENV" ~/.bashrc; then if ! grep -qxF "$TVM_HOME_ENV" ~/.bashrc; then
...@@ -112,14 +110,6 @@ else ...@@ -112,14 +110,6 @@ else
echo "PYTHONPATH is already set in ~/.bashrc" echo "PYTHONPATH is already set in ~/.bashrc"
fi fi
# Check and add the third line if not already present
if ! grep -qxF "$CUDA_DEVICE_ORDER_ENV" ~/.bashrc; then
echo "$CUDA_DEVICE_ORDER_ENV" >> ~/.bashrc
echo "Added CUDA_DEVICE_ORDER to ~/.bashrc"
else
echo "CUDA_DEVICE_ORDER is already set in ~/.bashrc"
fi
# Reload ~/.bashrc to apply the changes # Reload ~/.bashrc to apply the changes
source ~/.bashrc source ~/.bashrc
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment