diff --git a/src/deployment/jenkins/tvm_build_envs.py b/src/deployment/jenkins/tvm_build_envs.py index d3776fd81..341e75997 100644 --- a/src/deployment/jenkins/tvm_build_envs.py +++ b/src/deployment/jenkins/tvm_build_envs.py @@ -17,7 +17,7 @@ def _run(self, cmd): def create_envs(self): if len(self.frameworks) != 0: for framework in self.frameworks: - self._run(f'{self.conda_prefix}/bin/conda create -y --name tvm_{framework} --clone tvm_main_{self.branch}') + self._run(f'{self.conda_prefix}/bin/conda create -y --name tvm_{framework}_{self.branch} --clone tvm_main_{self.branch}') if framework != 'mxnet': self._run(f'{self.conda_prefix}/envs/tvm_{framework}_{self.branch}/bin/pip3 install {framework}') else: diff --git a/src/deployment/jenkins/tvm_build_pipeline.py b/src/deployment/jenkins/tvm_build_pipeline.py index 020d6874d..fbf2a9ec6 100644 --- a/src/deployment/jenkins/tvm_build_pipeline.py +++ b/src/deployment/jenkins/tvm_build_pipeline.py @@ -19,7 +19,7 @@ def build_tvm(self): self._run(f'{self.conda}/bin/conda install -n tvm_main_{self.branch} -c conda-forge -y gxx_linux-64') self._run(f'{self.conda}/envs/tvm_main_{self.branch}/bin/pip3 install -r requirements.txt') self._run(f'git clone --recursive https://github.com/apache/tvm -b {self.branch}') - self._run(f'cd tvm && mkdir -p build && cd build && cmake -DUSE_LLVM=ON ../ && make -j$(nproc --all) && cd ../python && {self.conda}/envs/tvm_main_{self.branch}/bin/python setup.py install --user') + self._run(f'cd tvm && mkdir -p build && cd build && cmake -DUSE_LLVM=ON -DUSE_BLAS=openblas ../ && make -j$(nproc --all) && cd ../python && {self.conda}/envs/tvm_main_{self.branch}/bin/python setup.py install --user') def cli_arguments_parse():