This commit is contained in:
Guolin Ke 2017-06-30 13:21:00 +08:00
Родитель 711a0a78c7
Коммит 9b435d651f
3 изменённых файлов: 4 добавлений и 11 удалений

Просмотреть файл

@ -59,7 +59,8 @@ if [[ ${TASK} == "gpu" ]]; then
export PATH="$AMDAPPSDK/include/:$PATH"
export BOOST_ROOT="$HOME/miniconda/"
LGB_VER=$(head -n 1 VERSION.txt)
cd $TRAVIS_BUILD_DIR/python-package && python setup.py sdist --gpu || exit -1
sed -i 's/std::string device_type = "cpu";/std::string device_type = "gpu";/' ../include/LightGBM/config.h
cd $TRAVIS_BUILD_DIR/python-package && python setup.py sdist || exit -1
cd $TRAVIS_BUILD_DIR/python-package/dist && pip install lightgbm-$LGB_VER.tar.gz -v --install-option=--gpu || exit -1
cd $TRAVIS_BUILD_DIR && pytest tests/python_package_test || exit -1
exit 0

Просмотреть файл

@ -1 +1 @@
2.0.3
2.0.4

Просмотреть файл

@ -104,16 +104,8 @@ class CustomInstall(install):
class CustomSdist(sdist):
user_options = sdist.user_options + [
('gpu', 'g', 'compile gpu version')
]
def initialize_options(self):
sdist.initialize_options(self)
self.gpu = 0
def run(self):
copy_files(use_gpu=self.gpu)
copy_files(use_gpu=True)
open("./_IS_SOURCE_PACKAGE.txt", 'w').close()
if os.path.exists("./lightgbm/Release/"):
shutil.rmtree('./lightgbm/Release/')