Skip to content

Commit

Permalink
Disable torch-mlir-core for release build (#2586)
Browse files Browse the repository at this point in the history
  • Loading branch information
vivekkhandelwal1 committed Nov 21, 2023
1 parent 647f2f5 commit b26797c
Showing 1 changed file with 6 additions and 17 deletions.
23 changes: 6 additions & 17 deletions .github/workflows/buildRelease.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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:

Expand Down Expand Up @@ -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:
Expand Down Expand Up @@ -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
Expand Down Expand Up @@ -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
Expand All @@ -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
Expand Down

0 comments on commit b26797c

Please sign in to comment.