* Make sure to install the nnvm_compiler library.

* Add conda packge.
This commit is contained in:
abergeron 2017-11-13 15:57:06 -05:00 коммит произвёл Tianqi Chen
Родитель c6c2e15663
Коммит a0b0546a29
3 изменённых файлов: 63 добавлений и 0 удалений

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

@ -113,8 +113,11 @@ if(BUILD_SHARED_NNVM)
if(WIN32)
install(TARGETS nnvm RUNTIME DESTINATION bin)
install(TARGETS nnvm ARCHIVE DESTINATION lib)
install(TARGETS nnvm_compiler RUNTIME DESTINATION bin)
install(TARGETS nnvm_compiler ARCHIVE DESTINATION lib)
else()
install(TARGETS nnvm LIBRARY DESTINATION lib)
install(TARGETS nnvm_compiler LIBRARY DESTINATION lib)
endif()
endif()
if(BUILD_STATIC_NNVM)

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

@ -0,0 +1,28 @@
#!/bin/bash
set -e
if [ -z "$PREFIX" ]; then
PREFIX="$CONDA_PREFIX"
fi
if [ "$(uname)" = 'Darwin' ]
then
# Without this, Apple's default shipped clang will refuse to see any
# headers like mutex.
export MACOSX_DEPLOYMENT_TARGET=10.9
fi
rm -rf build || true
mkdir -p build
cd build
# Enable static-libstdc++ to make it easier to link this library with
# other C++ compilers
CXXFLAGS=-static-libstdc++ cmake -DCMAKE_PREFIX_PATH=${PREFIX} -DCMAKE_INSTALL_PREFIX=${PREFIX} ..
make -j4 VERBOSE=1
make install/fast
cd ..
cd python
$PYTHON setup.py install
cd ..

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

@ -0,0 +1,32 @@
{% set version = "0.1.dev" %}
package:
name: nnvm
version: {{ version }}
source:
path: ../..
build:
number: 1
skip: True # [win]
requirements:
build:
- cmake
- python >=3
- numpy
- setuptools
- nose
- decorator
run:
- tvm
- topi
- python >=3
- numpy
- decorator
about:
home: https://github.com/dmlc/nnvm
license: Apache2
summary: Bring deep learning to bare metal