diff --git a/tests/test_tipc/llm/prepare.sh b/tests/test_tipc/llm/prepare.sh index 64333c6a1878..048738fd0b8f 100644 --- a/tests/test_tipc/llm/prepare.sh +++ b/tests/test_tipc/llm/prepare.sh @@ -20,4 +20,7 @@ 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 python setup_cuda.py install