Disable torch-mlir-core for release build (#2586)

pull/2576/head
Vivek Khandelwal 2023-11-21 09:06:14 +05:30 committed by GitHub
parent 647f2f5076
commit b26797c20b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 17 deletions

View File

@ -16,13 +16,8 @@ jobs:
runs-on: a100 runs-on: a100
strategy: strategy:
matrix: matrix:
package: [ torch-mlir, torch-mlir-core ] package: [ torch-mlir ]
py_version: [ cp38-cp38, cp311-cp311 ] py_version: [ cp38-cp38, cp311-cp311 ]
exclude:
- package: torch-mlir-core
py_version: cp38-cp38
- package: torch-mlir-core
py_version: cp310-cp310
steps: steps:
@ -90,7 +85,7 @@ jobs:
runs-on: linux-arm64 runs-on: linux-arm64
strategy: strategy:
matrix: matrix:
package: [ torch-mlir, torch-mlir-core ] package: [ torch-mlir ]
py_version: [ cp311-cp311 ] py_version: [ cp311-cp311 ]
steps: steps:
@ -159,7 +154,7 @@ jobs:
runs-on: macos-latest runs-on: macos-latest
strategy: strategy:
matrix: matrix:
package: [ torch-mlir, torch-mlir-core ] package: [ torch-mlir ]
steps: steps:
- name: Get torch-mlir - name: Get torch-mlir
uses: actions/checkout@v3 uses: actions/checkout@v3
@ -219,7 +214,7 @@ jobs:
runs-on: windows-latest runs-on: windows-latest
strategy: strategy:
matrix: matrix:
package: [ torch-mlir, torch-mlir-core ] package: [ torch-mlir ]
steps: steps:
- name: Get torch-mlir - name: Get torch-mlir
uses: actions/checkout@v3 uses: actions/checkout@v3
@ -235,14 +230,8 @@ jobs:
- name: Build Python wheels and smoke test. - name: Build Python wheels and smoke test.
shell: pwsh shell: pwsh
run: | run: |
if ( "${{ matrix.package }}" -eq "torch-mlir-core" )
{
$env:TORCH_MLIR_ENABLE_JIT_IR_IMPORTER='0'
$env:TORCH_MLIR_ENABLE_ONLY_MLIR_PYTHON_BINDINGS='1'
} else {
$env:TORCH_MLIR_ENABLE_JIT_IR_IMPORTER='1' $env:TORCH_MLIR_ENABLE_JIT_IR_IMPORTER='1'
$env:TORCH_MLIR_ENABLE_ONLY_MLIR_PYTHON_BINDINGS='0' $env:TORCH_MLIR_ENABLE_ONLY_MLIR_PYTHON_BINDINGS='0'
}
$env:TORCH_MLIR_PYTHON_PACKAGE_VERSION = '${{ github.event.inputs.python_package_version }}' $env:TORCH_MLIR_PYTHON_PACKAGE_VERSION = '${{ github.event.inputs.python_package_version }}'
./build_tools/python_deploy/build_windows.ps1 ./build_tools/python_deploy/build_windows.ps1