未验证 提交 e358ddac 编写于 作者: 6 6clc 提交者: GitHub

[CINN] set jobs by machine's cpu (#56605)

参照setup.py中设置max_jobs方法,设置cinn的build脚本的编译线程数。并且删除一些之前工作人员留下的对他们repo的依赖
上级 d7679426
......@@ -24,7 +24,10 @@ cinn_whl_path=python/dist/cinn-0.0.0-py3-none-any.whl
#export LLVM11_DIR=${workspace}/THIRDS/usr
JOBS=8
if [[ "" == ${JOBS} ]]; then
JOBS=`nproc`
fi
cuda_config=OFF
cudnn_config=OFF
......@@ -118,19 +121,6 @@ function prepare_model {
proxy_on
mkdir -p $build_dir/paddle
cd $build_dir/paddle
if [[ ! -f "libexternal_kernels.so.tgz" ]]; then
wget https://github.com/T8T9/files/raw/main/libexternal_kernels.so.tgz
fi
tar -zxvf libexternal_kernels.so.tgz
if [[ ! -f "paddle_1.8_fc_model.tgz" ]]; then
wget https://github.com/T8T9/files/raw/main/paddle_1.8_fc_model.tgz
fi
tar -zxvf paddle_1.8_fc_model.tgz
if [[ ! -f "mkldnn.tgz" ]]; then
wget https://github.com/T8T9/files/raw/main/mkldnn.tgz
fi
tar -zxvf mkldnn.tgz
cd $build_dir/third_party
python${py_version} $workspace/test/cinn/fake_model/naive_mul.py
python${py_version} $workspace/test/cinn/fake_model/naive_multi_fc.py
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册