fixes

itlab-vision · Jan 5, 2025 · c015fd2 · c015fd2
1 parent a56d536
commit c015fd2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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():