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