diff --git a/Jenkinsfile b/Jenkinsfile index 416469563bc3..a50b96d1942c 100644 --- a/Jenkinsfile +++ b/Jenkinsfile @@ -308,6 +308,24 @@ stage('Integration Test') { } } +/* +stage('Build packages') { + parallel 'conda CPU': { + node('CPU') { + sh "${docker_run} tvmai/conda-cpu ./conda/build_cpu.sh + } + }, + 'conda cuda': { + node('CPU') { + sh "${docker_run} tvmai/conda-cuda90 ./conda/build_cuda.sh + sh "${docker_run} tvmai/conda-cuda100 ./conda/build_cuda.sh + } + } + // Here we could upload the packages to anaconda for releases + // and/or the master branch +} +*/ + stage('Deploy') { node('doc') { ws('workspace/tvm/deploy-docs') { diff --git a/conda/Dockerfile.template b/conda/Dockerfile.template index 59b9ac96814e..1b5dc6fbef5e 100644 --- a/conda/Dockerfile.template +++ b/conda/Dockerfile.template @@ -15,9 +15,13 @@ # specific language governing permissions and limitations # under the License. -FROM nvidia/cuda:{{ cuda_version }}-devel-centos6 +FROM nvidia/cuda:{{ cuda_version }}-devel-ubuntu16.04 -RUN curl -fsSL http://developer.download.nvidia.com/compute/redist/cudnn/v{{ cudnn_short_version }}/cudnn-{{ cuda_version }}-linux-x64-v{{ cudnn_version }}.tgz -O && \ +RUN apt-get update && apt-get install -y --no-install-recommends \ + bzip2 curl sudo binutils && \ + rm -rf /var/lib/apt/lists/* + +RUN curl -fsSL http://developer.download.nvidia.com/compute/redist/cudnn/v{{ cudnn_short_version }}/cudnn-{{ cuda_version }}-linux-x64-v{{ cudnn_version }}.tgz -O && \ tar --no-same-owner -xzf cudnn-{{ cuda_version }}-linux-x64-v{{ cudnn_version }}.tgz -C /usr/local && \ rm cudnn-{{ cuda_version }}-linux-x64-v{{ cudnn_version }}.tgz && \ ldconfig @@ -27,13 +31,16 @@ RUN curl -o ~/miniconda.sh -O https://repo.continuum.io/miniconda/Miniconda3-lat chmod +x ~/miniconda.sh && \ ~/miniconda.sh -b -p /opt/conda && \ rm ~/miniconda.sh && \ + /opt/conda/bin/conda upgrade --all && \ /opt/conda/bin/conda install conda-build conda-verify && \ /opt/conda/bin/conda clean -ya +RUN /opt/conda/bin/conda install --download-only cmake make zlib +RUN /opt/conda/bin/conda install --download-only -c numba llvmdev=8.0.0 + ENV PATH /opt/conda/bin:$PATH ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64 +ENV CONDA_BLD_PATH /tmp WORKDIR /workspace RUN chmod -R a+w /workspace - -CMD conda build --output-folder /workspace/conda/pkg --variants '{cuda: True, cuda_version: {{ cuda_version }}}' /workspace/conda/tvm-libs diff --git a/conda/Makefile b/conda/Makefile deleted file mode 100644 index cda546ac73ce..000000000000 --- a/conda/Makefile +++ /dev/null @@ -1,22 +0,0 @@ -# Licensed to the Apache Software Foundation (ASF) under one -# or more contributor license agreements. See the NOTICE file -# distributed with this work for additional information -# regarding copyright ownership. The ASF licenses this file -# to you under the Apache License, Version 2.0 (the -# "License"); you may not use this file except in compliance -# with the License. You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, -# software distributed under the License is distributed on an -# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY -# KIND, either express or implied. See the License for the -# specific language governing permissions and limitations -# under the License. - -packages: - conda build tvm-libs - conda build tvm - conda build topi - conda built nnvm diff --git a/conda/nnvm/build.sh b/conda/build_cpu.sh old mode 100644 new mode 100755 similarity index 68% rename from conda/nnvm/build.sh rename to conda/build_cpu.sh index bdd333f57734..992b1a369b96 --- a/conda/nnvm/build.sh +++ b/conda/build_cpu.sh @@ -1,4 +1,4 @@ -#!/bin/bash +#!/bin/sh # Licensed to the Apache Software Foundation (ASF) under one # or more contributor license agreements. See the NOTICE file # distributed with this work for additional information @@ -6,9 +6,9 @@ # to you under the Apache License, Version 2.0 (the # "License"); you may not use this file except in compliance # with the License. You may obtain a copy of the License at -# +# # http://www.apache.org/licenses/LICENSE-2.0 -# +# # Unless required by applicable law or agreed to in writing, # software distributed under the License is distributed on an # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY @@ -17,6 +17,15 @@ # under the License. set -e +set -u -cd nnvm/python -$PYTHON setup.py install --single-version-externally-managed --record=/tmp/record.txt +# This is a fix for a weird bug in conda that makes it think +# it can't write in /tmp +HOME=/tmp +mkdir -p /tmp/.conda/pkgs +touch /tmp/.conda/pkgs/urls.txt +touch /tmp/.conda/environments.txt + + +conda build --output-folder=conda/pkg -c numba conda/tvm-libs +conda build --output-folder=conda/pkg -m conda/conda_build_config.yaml conda/tvm diff --git a/conda/topi/build.sh b/conda/build_cuda.sh old mode 100644 new mode 100755 similarity index 70% rename from conda/topi/build.sh rename to conda/build_cuda.sh index 4e5aafb93766..2c9a20ae66ae --- a/conda/topi/build.sh +++ b/conda/build_cuda.sh @@ -1,4 +1,4 @@ -#!/bin/bash +#!/bin/sh # Licensed to the Apache Software Foundation (ASF) under one # or more contributor license agreements. See the NOTICE file # distributed with this work for additional information @@ -6,9 +6,9 @@ # to you under the Apache License, Version 2.0 (the # "License"); you may not use this file except in compliance # with the License. You may obtain a copy of the License at -# +# # http://www.apache.org/licenses/LICENSE-2.0 -# +# # Unless required by applicable law or agreed to in writing, # software distributed under the License is distributed on an # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY @@ -17,6 +17,14 @@ # under the License. set -e +set -u -cd topi/python -$PYTHON setup.py install --single-version-externally-managed --record=/tmp/record.txt +# This is a fix for a weird bug in conda that makes it think +# it can't write in /tmp +HOME=/tmp +mkdir -p /tmp/.conda/pkgs +touch /tmp/.conda/pkgs/urls.txt +touch /tmp/.conda/environments.txt + + +conda build --output-folder=conda/pkg --variants "{cuda: True, cuda_version: ${CUDA_VERSION%.*}}" -c numba conda/tvm-libs diff --git a/conda/topi/meta.yaml b/conda/cross-linux.cmake similarity index 54% rename from conda/topi/meta.yaml rename to conda/cross-linux.cmake index f4bc8950d4c4..360400267ae0 100644 --- a/conda/topi/meta.yaml +++ b/conda/cross-linux.cmake @@ -15,37 +15,24 @@ # specific language governing permissions and limitations # under the License. -{% set version = "0.6.dev" %} +# this one is important +set(CMAKE_SYSTEM_NAME Linux) +set(CMAKE_PLATFORM Linux) +#this one not so much +set(CMAKE_SYSTEM_VERSION 1) -package: - name: topi - version: {{ version }} +# specify the cross compiler +set(CMAKE_C_COMPILER $ENV{CC}) -source: - path: ../.. +# where is the target environment +set(CMAKE_FIND_ROOT_PATH $ENV{PREFIX} $ENV{BUILD_PREFIX}/$ENV{HOST}/sysroot) -build: - number: 1 +# search for programs in the build host directories +set(CMAKE_FIND_ROOT_PATH_MODE_PROGRAM NEVER) -requirements: - host: - - python {{ python }} - - numpy - - setuptools - - decorator - - tvm-libs =={{ version }} - run: - - python - - {{ pin_compatible('numpy') }} - - decorator - - tvm-libs =={{ version }} - - tvm =={{ version }} +# for libraries and headers in the target directories +set(CMAKE_FIND_ROOT_PATH_MODE_LIBRARY ONLY) +set(CMAKE_FIND_ROOT_PATH_MODE_INCLUDE ONLY) -test: - imports: - - topi - -about: - home: https://github.com/dmlc/tvm - license: Apache2 - summary: "TOPI: TVM Operator Inventory" +# god-awful hack because it seems to not run correct tests to determine this: +set(__CHAR_UNSIGNED___EXITCODE 1) diff --git a/conda/nnvm/meta.yaml b/conda/nnvm/meta.yaml deleted file mode 100644 index d948484a61e5..000000000000 --- a/conda/nnvm/meta.yaml +++ /dev/null @@ -1,56 +0,0 @@ -# Licensed to the Apache Software Foundation (ASF) under one -# or more contributor license agreements. See the NOTICE file -# distributed with this work for additional information -# regarding copyright ownership. The ASF licenses this file -# to you under the Apache License, Version 2.0 (the -# "License"); you may not use this file except in compliance -# with the License. You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, -# software distributed under the License is distributed on an -# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY -# KIND, either express or implied. See the License for the -# specific language governing permissions and limitations -# under the License. - -{% set version = "0.6.dev" %} - -package: - name: nnvm - version: {{ version }} - -source: - path: ../.. - -build: - number: 1 - skip: True # [win] - -requirements: - build: - - {{ compiler('cxx') }} - host: - - python {{ python }} - - cython - - numpy - - setuptools - - decorator - - tvm-libs =={{ version }} - run: - - tvm =={{ version }} - - topi =={{ version }} - - tvm-libs =={{ version }} - - python - - {{ pin_compatible('numpy') }} - - decorator - -test: - imports: - - nnvm - -about: - home: https://github.com/dmlc/nnvm - license: Apache2 - summary: Bring deep learning to bare metal diff --git a/conda/build_cuda.py b/conda/render_cuda.py similarity index 74% rename from conda/build_cuda.py rename to conda/render_cuda.py index 47af6ce4564e..8057892fd83c 100644 --- a/conda/build_cuda.py +++ b/conda/render_cuda.py @@ -29,8 +29,8 @@ # and from conda. # These two must be in sync -CUDNN_FULL_VERSION = '7.3.1.20' -CUDNN_VERSION = '7.3.1' +CUDNN_FULL_VERSION = '7.6.0.64' +CUDNN_VERSION = '7.6.0' condadir = os.path.dirname(sys.argv[0]) @@ -47,30 +47,15 @@ def render_dockerfile(version): cudnn_short_version=CUDNN_VERSION, cudnn_version=CUDNN_FULL_VERSION) fname = os.path.join(condadir, - 'Dockerfile.cuda' + version.replace('.', '')) + '../docker/Dockerfile.conda_cuda' + version.replace('.', '')) with open(fname, 'w') as f: f.write(txt) return fname -def build_docker(version): - vv = version.replace('.', '') - fname = render_dockerfile(version) - tagname = f'tvm-cuda{ vv }-forge' - subprocess.run(['docker', 'build', '-t', tagname, - condadir, '-f', fname], check=True) - return tagname - - -def build_pkg(version): - tagname = build_docker(version) - subprocess.run(['docker', 'run', '--rm', '-v', f'{ srcdir }:/workspace', - tagname], check=True) - - if __name__ == '__main__': build_versions = CUDA_VERSIONS if len(sys.argv) > 1: build_versions = sys.argv[1:] for version in build_versions: - build_pkg(version) + render_dockerfile(version) diff --git a/conda/tvm-libs/build.sh b/conda/tvm-libs/build.sh index e0b85910475e..94919c60e779 100644 --- a/conda/tvm-libs/build.sh +++ b/conda/tvm-libs/build.sh @@ -17,24 +17,37 @@ # under the License. set -e - -if [ "$cuda" == "True" ]; then - CUDA_OPT="-DUSE_CUDA=ON -DUSE_CUBLAS=ON -DUSE_CUDNN=ON" -else - CUDA_OPT="" -fi +set -u if [ "$target_platform" == "osx-64" ]; then # macOS 64 bits - METAL_OPT="" # Conda can only target 10.9 for now + METAL_OPT="-DUSE_METAL=ON" + TOOLCHAIN_OPT="-DCMAKE_OSX_DEPLOYMENT_TARGET=10.11" else METAL_OPT="" + if [ "$target_platform" == "linux-64" ]; then + # Linux 64 bits + TOOLCHAIN_OPT="-DCMAKE_TOOLCHAIN_FILE=${RECIPE_DIR}/../cross-linux.cmake" + else + # Windows (or 32 bits, which we don't support) + TOOLCHAIN_OPT="" + fi +fi + +# When cuda is not set, we default to False +cuda=${cuda:-False} + +if [ "$cuda" == "True" ]; then + CUDA_OPT="-DUSE_CUDA=ON -DUSE_CUBLAS=ON -DUSE_CUDNN=ON" + TOOLCHAIN_OPT="" +else + CUDA_OPT="" fi rm -rf build || true mkdir -p build cd build -cmake $METAL_OPT $CUDA_OPT -DUSE_LLVM=$PREFIX/bin/llvm-config -DINSTALL_DEV=ON -DCMAKE_INSTALL_PREFIX="$PREFIX" .. +cmake $METAL_OPT $CUDA_OPT -DUSE_LLVM=$PREFIX/bin/llvm-config -DINSTALL_DEV=ON -DCMAKE_INSTALL_PREFIX="$PREFIX" $TOOLCHAIN_OPT .. make -j${CPU_COUNT} VERBOSE=1 make install cd .. diff --git a/conda/tvm-libs/meta.yaml b/conda/tvm-libs/meta.yaml index aad8f251c2a6..e3422a2174ef 100644 --- a/conda/tvm-libs/meta.yaml +++ b/conda/tvm-libs/meta.yaml @@ -15,7 +15,7 @@ # specific language governing permissions and limitations # under the License. -{% set version = "0.6.dev" %} +{% set version = "0.6.dev1" %} package: name: tvm-libs @@ -25,21 +25,22 @@ source: path: ../.. build: - number: 1 - string: cuda{{ cuda_version }}_{{ PKG_BUILDNUM }} # [cuda] + number: 0 + string: cuda{{ cuda_version | replace('.', '') }}h{{ PKG_HASH }}_{{ PKG_BUILDNUM }} # [cuda] requirements: build: - # The OS X build will require some manual setup or it will break - # See https://docs.conda.io/projects/conda-build/en/latest/source/resources/compiler-tools.html#macos-sdk - - {{ compiler('cxx') }} - host: + # The anaconda compilers for OS X are old an annoying + # so we rely on the platform ones for now + - {{ compiler('cxx') }} # [linux] - cmake - - llvmdev ==6.0.0 + - make + host: + - llvmdev ==8.0.0 - zlib # [linux] run: - {{ pin_compatible('cudatoolkit', lower_bound=cuda_version, max_pin='x.x') }} # [cuda] - - {{ pin_compatible('cudnn', lower_bound='7.3.1', max_pin='x') }} # [cuda] + - {{ pin_compatible('cudnn', lower_bound='7.6.0', max_pin='x') }} # [cuda] about: home: https://github.com/dmlc/tvm diff --git a/conda/tvm/build.sh b/conda/tvm/build.sh index 6626aa592091..494f90f0afa0 100644 --- a/conda/tvm/build.sh +++ b/conda/tvm/build.sh @@ -17,6 +17,16 @@ # under the License. set -e +set -u cd python $PYTHON setup.py install --single-version-externally-managed --record=/tmp/record.txt +cd .. + +cd topi/python +$PYTHON setup.py install --single-version-externally-managed --record=/tmp/record.txt +cd ../.. + +cd nnvm/python +$PYTHON setup.py install --single-version-externally-managed --record=/tmp/record.txt +cd ../.. diff --git a/conda/tvm/meta.yaml b/conda/tvm/meta.yaml index 221dc7950f75..0daca4bcea2b 100644 --- a/conda/tvm/meta.yaml +++ b/conda/tvm/meta.yaml @@ -15,7 +15,7 @@ # specific language governing permissions and limitations # under the License. -{% set version = "0.6.dev" %} +{% set version = "0.6.dev1" %} package: name: tvm @@ -25,7 +25,7 @@ source: path: ../.. build: - number: 1 + number: 0 requirements: build: @@ -46,6 +46,15 @@ requirements: test: imports: - tvm + - topi + - nnvm + requires: + - nose + - scipy + source_files: + - tests/python + commands: + - python -m nose -v tests/python/integration about: home: https://github.com/dmlc/tvm diff --git a/docker/Dockerfile.conda_cpu b/docker/Dockerfile.conda_cpu new file mode 100644 index 000000000000..0660b5daa0e2 --- /dev/null +++ b/docker/Dockerfile.conda_cpu @@ -0,0 +1,41 @@ +# Licensed to the Apache Software Foundation (ASF) under one +# or more contributor license agreements. See the NOTICE file +# distributed with this work for additional information +# regarding copyright ownership. The ASF licenses this file +# to you under the Apache License, Version 2.0 (the +# "License"); you may not use this file except in compliance +# with the License. You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, +# software distributed under the License is distributed on an +# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY +# KIND, either express or implied. See the License for the +# specific language governing permissions and limitations +# under the License. + +FROM ubuntu:16.04 + +RUN apt-get update && apt-get install -y bzip2 curl sudo binutils && rm -rf /var/lib/apt/lists/* + +RUN curl -o ~/miniconda.sh -O https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh && \ + chmod +x ~/miniconda.sh && \ + ~/miniconda.sh -b -p /opt/conda && \ + rm ~/miniconda.sh && \ + /opt/conda/bin/conda upgrade --all && \ + /opt/conda/bin/conda install conda-build conda-verify && \ + /opt/conda/bin/conda clean -ya + +# Cache some of the packages for the builds +RUN /opt/conda/bin/conda install --download-only cmake make zlib && \ + /opt/conda/bin/conda install --download-only -c numba llvmdev=8.0.0 && \ + /opt/conda/bin/conda create -n py35 --download-only nose scipy numpy=1.11 cython decorator python=3.5 && \ + /opt/conda/bin/conda create -n py36 --download-only nose scipy numpy=1.11 cython decorator python=3.6 && \ + /opt/conda/bin/conda create -n py37 --download-only nose scipy numpy=1.11 cython decorator python=3.7 + +ENV PATH /opt/conda/bin:$PATH +ENV CONDA_BLD_PATH /tmp + +WORKDIR /workspace +RUN chmod -R a+w /workspace diff --git a/docker/Dockerfile.conda_cuda100 b/docker/Dockerfile.conda_cuda100 new file mode 100644 index 000000000000..d6e1cddbfd37 --- /dev/null +++ b/docker/Dockerfile.conda_cuda100 @@ -0,0 +1,46 @@ +# Licensed to the Apache Software Foundation (ASF) under one +# or more contributor license agreements. See the NOTICE file +# distributed with this work for additional information +# regarding copyright ownership. The ASF licenses this file +# to you under the Apache License, Version 2.0 (the +# "License"); you may not use this file except in compliance +# with the License. You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, +# software distributed under the License is distributed on an +# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY +# KIND, either express or implied. See the License for the +# specific language governing permissions and limitations +# under the License. + +FROM nvidia/cuda:10.0-devel-ubuntu16.04 + +RUN apt-get update && apt-get install -y --no-install-recommends \ + bzip2 curl sudo binutils && \ + rm -rf /var/lib/apt/lists/* + +RUN curl -fsSL http://developer.download.nvidia.com/compute/redist/cudnn/v7.6.0/cudnn-10.0-linux-x64-v7.6.0.64.tgz -O && \ + tar --no-same-owner -xzf cudnn-10.0-linux-x64-v7.6.0.64.tgz -C /usr/local && \ + rm cudnn-10.0-linux-x64-v7.6.0.64.tgz && \ + ldconfig + + +RUN curl -o ~/miniconda.sh -O https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh && \ + chmod +x ~/miniconda.sh && \ + ~/miniconda.sh -b -p /opt/conda && \ + rm ~/miniconda.sh && \ + /opt/conda/bin/conda upgrade --all && \ + /opt/conda/bin/conda install conda-build conda-verify && \ + /opt/conda/bin/conda clean -ya + +RUN /opt/conda/bin/conda install --download-only cmake make zlib +RUN /opt/conda/bin/conda install --download-only -c numba llvmdev=8.0.0 + +ENV PATH /opt/conda/bin:$PATH +ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64 +ENV CONDA_BLD_PATH /tmp + +WORKDIR /workspace +RUN chmod -R a+w /workspace \ No newline at end of file diff --git a/docker/Dockerfile.conda_cuda90 b/docker/Dockerfile.conda_cuda90 new file mode 100644 index 000000000000..f55aa1bf2e12 --- /dev/null +++ b/docker/Dockerfile.conda_cuda90 @@ -0,0 +1,46 @@ +# Licensed to the Apache Software Foundation (ASF) under one +# or more contributor license agreements. See the NOTICE file +# distributed with this work for additional information +# regarding copyright ownership. The ASF licenses this file +# to you under the Apache License, Version 2.0 (the +# "License"); you may not use this file except in compliance +# with the License. You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, +# software distributed under the License is distributed on an +# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY +# KIND, either express or implied. See the License for the +# specific language governing permissions and limitations +# under the License. + +FROM nvidia/cuda:9.0-devel-ubuntu16.04 + +RUN apt-get update && apt-get install -y --no-install-recommends \ + bzip2 curl sudo binutils && \ + rm -rf /var/lib/apt/lists/* + +RUN curl -fsSL http://developer.download.nvidia.com/compute/redist/cudnn/v7.6.0/cudnn-9.0-linux-x64-v7.6.0.64.tgz -O && \ + tar --no-same-owner -xzf cudnn-9.0-linux-x64-v7.6.0.64.tgz -C /usr/local && \ + rm cudnn-9.0-linux-x64-v7.6.0.64.tgz && \ + ldconfig + + +RUN curl -o ~/miniconda.sh -O https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh && \ + chmod +x ~/miniconda.sh && \ + ~/miniconda.sh -b -p /opt/conda && \ + rm ~/miniconda.sh && \ + /opt/conda/bin/conda upgrade --all && \ + /opt/conda/bin/conda install conda-build conda-verify && \ + /opt/conda/bin/conda clean -ya + +RUN /opt/conda/bin/conda install --download-only cmake make zlib +RUN /opt/conda/bin/conda install --download-only -c numba llvmdev=8.0.0 + +ENV PATH /opt/conda/bin:$PATH +ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64 +ENV CONDA_BLD_PATH /tmp + +WORKDIR /workspace +RUN chmod -R a+w /workspace \ No newline at end of file diff --git a/version.py b/version.py index 1df897a40fb4..c949d5c2ead2 100644 --- a/version.py +++ b/version.py @@ -24,8 +24,6 @@ - tvm-root/include/tvm/runtime/c_runtime_api.h - tvm-root/web/tvm_runtime.js - tvm-root/conda/tvm/meta.yaml -- tvm-root/conda/topi/meta.yaml -- tvm-root/conda/nnvm/meta.yaml - tvm-root/conda/tvm-libs/meta.yaml """ import os @@ -71,7 +69,7 @@ def main(): update(os.path.join(proj_root, "include", "tvm", "runtime", "c_runtime_api.h"), "(?<=TVM_VERSION \")[.0-9a-z]+", __version__) # conda - for path in ["tvm", "topi", "nnvm", "tvm-libs"]: + for path in ["tvm", "tvm-libs"]: update(os.path.join(proj_root, "conda", path, "meta.yaml"), "(?<=version = \")[.0-9a-z]+", __version__) # web