Skip to content

Commit

Permalink
Bump to depend on iree 3.0.0 packages. (#275)
Browse files Browse the repository at this point in the history
See the release tracker at
iree-org/iree#19063.
  • Loading branch information
ScottTodd authored Nov 18, 2024
1 parent ff71682 commit 583004f
Show file tree
Hide file tree
Showing 7 changed files with 16 additions and 8 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/ci-tk.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,7 @@ jobs:
# from non default locations first. Installing the PyTorch CPU
# wheels saves multiple minutes and a lot of bandwidth on runner setup.
pip install --no-compile -r pytorch-cpu-requirements.txt
pip install --no-cache-dir -r iree-requirements-ci.txt
pip install --no-cache-dir -r iree-requirements-ci.txt --upgrade
pip install -r requirements.txt -e .
- name: Run unit tests
Expand Down
2 changes: 1 addition & 1 deletion .github/workflows/perf.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@ jobs:
# from non default locations first. Installing the PyTorch CPU
# wheels saves multiple minutes and a lot of bandwidth on runner setup.
pip install --no-compile -r pytorch-rocm-requirements.txt
pip install --no-cache-dir -r iree-requirements-ci.txt
pip install --no-cache-dir -r iree-requirements-ci.txt --upgrade
pip install -r requirements.txt -e .
- name: Run e2e tests on MI300
Expand Down
3 changes: 2 additions & 1 deletion build_tools/build_release.py
Original file line number Diff line number Diff line change
Expand Up @@ -72,7 +72,8 @@ def write_version_info(args):
info_dict["package-version"] = package_version

with open(VERSION_INFO_FILE, "wt") as f:
json.dump(info_dict, f)
json.dump(info_dict, f, indent=2)
f.write("\n")

print(f"Updated version_info.json:\n{json.dumps(info_dict, indent=2)}")

Expand Down
6 changes: 6 additions & 0 deletions docs/releasing.md
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,12 @@ https://pypi.org/project/iree-base-runtime/ packages. Releases can either be
conducted independently, or they can be coordinated across projects by
initiating a release here.

## Start with a clean test directory

```bash
rm -rf wheelhouse/
```

## Building Artifacts

Build a pre-release:
Expand Down
5 changes: 3 additions & 2 deletions iree-requirements-ci.txt
Original file line number Diff line number Diff line change
Expand Up @@ -5,9 +5,10 @@

# Uncomment to select a nightly version.
# --find-links https://iree.dev/pip-release-links.html
# --pre

# Uncomment to skip versions from PyPI (so _only_ nightly versions).
# --no-index

iree-base-compiler<=2.9.0
iree-base-runtime<=2.9.0
iree-base-compiler<=3.0.0
iree-base-runtime<=3.0.0
4 changes: 2 additions & 2 deletions iree/turbine/kernel/compiler/dispatch_codegen.py
Original file line number Diff line number Diff line change
Expand Up @@ -165,8 +165,8 @@ def abi_type(binding: BindingDesc):
def_func_args = list(def_func_block.arguments)
if workgroup_size is not None and subgroup_size is not None:
def_func_op.attributes["translation_info"] = Attribute.parse(
f"#iree_codegen.translation_info<None "
f"workgroup_size=[{','.join(str(x) for x in workgroup_size)}]"
f"#iree_codegen.translation_info<pipeline = None "
f"workgroup_size=[{','.join(str(x) for x in workgroup_size)}] "
f"subgroup_size={subgroup_size}>"
)

Expand Down
2 changes: 1 addition & 1 deletion version_info.json
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
{
"package-version": "3.0.0"
"package-version": "3.0.0"
}

0 comments on commit 583004f

Please sign in to comment.