Skip to content

Commit

Permalink
no need for rust from main
Browse files Browse the repository at this point in the history
  • Loading branch information
IlyasMoutawwakil committed Feb 19, 2025
1 parent 007ddbe commit 943f67e
Showing 1 changed file with 6 additions and 14 deletions.
20 changes: 6 additions & 14 deletions tests/ci/accelerate.sh
Original file line number Diff line number Diff line change
Expand Up @@ -5,20 +5,13 @@ echo "HABANA_VISIBLE_DEVICES=${HABANA_VISIBLE_DEVICES}"
echo "HABANA_VISIBLE_MODULES=${HABANA_VISIBLE_MODULES}"

# Install Accelerate and DeepSpeed
git clone https://github.com/huggingface/accelerate.git --branch hpu-support --depth 1
git clone https://github.com/huggingface/accelerate.git
cd accelerate
pip install -e .[testing] git+https://github.com/HabanaAI/[email protected]

# Install Rust and build safetensors
curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
. "$HOME/.cargo/env"
rustup update
git clone https://github.com/huggingface/safetensors
cd safetensors
git checkout fa833511664338bfc927fc02653ddb7d38d40be9
pip install setuptools_rust
pip install -e bindings/python
cd ..
git checkout hpu-support
git pull
pip install -e .[testing]
pip install git+https://github.com/HabanaAI/[email protected]


# Set environment variables
export PT_ENABLE_INT64_SUPPORT=1
Expand All @@ -39,7 +32,6 @@ if [ $? -ne 0 ]; then
exit 1
fi


echo "Running Core tests"
make test_core
if [ $? -ne 0 ]; then
Expand Down

0 comments on commit 943f67e

Please sign in to comment.