update tvm to v0.15.0

microsoft · Jan 29, 2024 · 122e256 · 122e256
1 parent 2036a4a
commit 122e256
Showing 1 changed file with 29 additions and 26 deletions.
diff --git a/.github/workflows/pythonapp.yml b/.github/workflows/pythonapp.yml
@@ -25,6 +25,8 @@ jobs:
       matrix:
         os: [ubuntu-22.04, macos-12, windows-2022]
         python-version: ['3.8', '3.9', '3.10', '3.11']
+    env:
+      TVM_VERSION_TAG: v0.15.0
 
     steps:
     - uses: actions/checkout@v4
@@ -73,54 +75,55 @@ jobs:
       run: |
         python -m pip install .[extra,onnx,sparkml]
         python -m pip install pandas
-    - name: Install TVM from pypi if Ubuntu and not python 3.11
-      if: ${{ startsWith(matrix.os, 'ubuntu') && matrix.python-version != '3.11' }}
-      run: python -m pip install apache-tvm==0.10.0
     - uses: actions/cache@v3
       # TVM takes forever, we try to cache it.
-      if: ${{ startsWith(matrix.os, 'macos') && matrix.python-version != '3.11' }}
+      if: ${{ startsWith(matrix.os, 'macos') || startsWith(matrix.os, 'ubuntu')}}
       id: cache
       env:
         CACHE_NUMBER: 9
       with:
         path: ~/work/hummingbird/tvm
-        key: ${{ matrix.os }}-${{ env.CACHE_NUMBER }}-${{ matrix.python-version }}-tvm-0.10
+        key: ${{ matrix.os }}-python${{ matrix.python-version }}-tvm-${{ env.TVM_VERSION_TAG }}-${{ env.CACHE_NUMBER }}
     # Getting TVM requires: 1) Install LLVM 2) fetching TVM from github, 3) cmake, 4) make, 5) install python dependency.
     # 2 to 4 will be retrieved from the cache.
     # The pipeline only works for Unix systems. For windows we will have to compile LLVM from source which is a no go.
+    - name: Install LLVM if Ubuntu
+      if: ${{ startsWith(matrix.os, 'ubuntu')}}
+      run: |
+        sudo apt install -y llvm-14
     - name: Install LLVM if Mac
-      if: ${{ startsWith(matrix.os, 'macos') && matrix.python-version != '3.11'}}
+      if: ${{ startsWith(matrix.os, 'macos')}}
       run: |
         brew install llvm@14
-    - name: Fetch and prepare TVM for compilation if Mac
-      if: ${{ steps.cache.outputs.cache-hit != 'true' && startsWith(matrix.os, 'macos') && matrix.python-version != '3.11' }}
+    - name: Fetch and prepare TVM for compilation if Mac or Ubuntu
+      if: ${{ steps.cache.outputs.cache-hit != 'true' && (startsWith(matrix.os, 'macos') || startsWith(matrix.os, 'ubuntu')) }}
       run: |
         cd ~/work/hummingbird
         git clone https://github.com/apache/tvm.git
         cd tvm
-        git checkout tags/v0.10.0
+        git checkout tags/${{env.TVM_VERSION_TAG}}
         git submodule update --recursive --init
         cmake -E make_directory build
     - name: CMake TVM if Mac
-      if: ${{ steps.cache.outputs.cache-hit != 'true' && startsWith(matrix.os, 'macos') && matrix.python-version != '3.11' }}
-      working-directory: ../tvm/build
-      run: >-
-        MACOSX_DEPLOYMENT_TARGET=10.13 cmake
-        "-DUSE_RPC=ON"
-        "-DUSE_GRAPH_RUNTIME=ON"
-        "-DUSE_LLVM=$(brew --prefix llvm@14)/bin/llvm-config --link-static"
-        ..
-    - name: Build TVM if Mac
-      if: ${{ steps.cache.outputs.cache-hit != 'true' && startsWith(matrix.os, 'macos') && matrix.python-version != '3.11' }}
-      working-directory: ../tvm/build
+      if: ${{ steps.cache.outputs.cache-hit != 'true' && startsWith(matrix.os, 'macos') }}
+      run: |
+        cd ~/work/hummingbird/tvm
+        cmake -S . -B build -DUSE_RPC=ON -DUSE_GRAPH_RUNTIME=ON -DUSE_LLVM=$(brew --prefix llvm@14)/bin/llvm-config
+    - name: CMake TVM if Ubuntu
+      if: ${{ steps.cache.outputs.cache-hit != 'true' && startsWith(matrix.os, 'ubuntu') }}
+      run: |
+        cd ~/work/hummingbird/tvm
+        cmake -S . -B build -DUSE_RPC=ON -DUSE_GRAPH_RUNTIME=ON -DUSE_LLVM=llvm-config-14
+    - name: Build TVM if Mac or Ubuntu
+      if: ${{ steps.cache.outputs.cache-hit != 'true' && (startsWith(matrix.os, 'macos') || startsWith(matrix.os, 'ubuntu')) }}
       run: |
-        make -j3
-    - name: Install python TVM if Mac and not python 3.11
-      if: ${{  startsWith(matrix.os, 'macos') && matrix.python-version != '3.11' }}
-      working-directory: ../tvm/python
+        cd ~/work/hummingbird/tvm
+        cmake --build build
+    - name: Install python TVM if Mac or Ubuntu
+      if: ${{  (startsWith(matrix.os, 'macos') || startsWith(matrix.os, 'ubuntu')) }}
       run: |
+        cd ~/work/hummingbird/tvm/python
         python -m pip install -U wheel packaging
-        python -m pip install cython==0.29.34
         python setup.py bdist_wheel
         python -m pip install dist/tvm-*.whl
 
@@ -132,7 +135,7 @@ jobs:
     - name: Coverage 3.9 Linux
       if: ${{ matrix.python-version == '3.9' && startsWith(matrix.os, 'ubuntu') }}
       run: |
-        coverage run -a -m pytest tests
+        coverage run -a -m pytest tests -v
         coverage xml
     - name: Upload coverage to Codecov
       if: ${{ matrix.python-version == '3.9' && startsWith(matrix.os, 'ubuntu') }}