Skip to content

Commit

Permalink
Fix llm ce predict run error (#9149)
Browse files Browse the repository at this point in the history
* fix llm ce predict run error
  • Loading branch information
fightfat authored Sep 19, 2024
1 parent a6cf882 commit 7faad55
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/test_tipc/llm/prepare.sh
Original file line number Diff line number Diff line change
Expand Up @@ -14,13 +14,13 @@

cd ..
sed -i -e "s/paddlepaddle/#paddlepaddle/g" requirements-dev.txt
sed -i -e "s/pip install paddlepaddle/#pip install paddlepaddle/g" Makefile
sed -i -e "s/pip install --pre paddlepaddle/#pip install --pre paddlepaddle/g" Makefile

make install

cd ./csrc
pip install -r requirements.txt
wget https://paddle-qa.bj.bcebos.com/benchmark/PaddleNLP/cutlass.tar
tar -xvf cutlass.tar
mv cutlass ./gpu/cutlass_kernels
mv cutlass ./third_party/cutlass
python setup_cuda.py install

0 comments on commit 7faad55

Please sign in to comment.