Skip to content

Commit

Permalink
Adapt to iree-turbine naming changes. (#11)
Browse files Browse the repository at this point in the history
This commit adapts to the recent changes made to the packaging naming in
the iree-org/iree-turbine repo:
iree-org/iree-turbine#197
  • Loading branch information
saienduri authored Oct 6, 2024
1 parent acec7b4 commit bf37c5c
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 6 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/run_bench.yml
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@ jobs:
pip install --find-links https://iree.dev/pip-release-links.html iree-compiler iree-runtime --upgrade
pip install -r requirements.txt
pip install --no-compile --pre --upgrade -e common_tools
pip install shark-turbine@git+https://github.com/iree-org/iree-turbine.git@main
pip install iree-turbine@git+https://github.com/iree-org/iree-turbine.git@main
- name: Convolutions
run: |
Expand Down
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ python3.11 -m venv bench_venv
source bench_venv/bin/activate
pip install -r requirements.txt
pip install --no-compile --pre --upgrade -e common_tools
pip install shark-turbine@git+https://github.com/iree-org/iree-turbine.git@main
pip install iree-turbine@git+https://github.com/iree-org/iree-turbine.git@main
```

## Performance
Expand Down
8 changes: 4 additions & 4 deletions gemmbench/gemm_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,10 +2,10 @@
from dataclasses import dataclass
from pathlib import Path
from typing import Optional
import shark_turbine.kernel as tk
import shark_turbine.kernel.lang as tkl
import shark_turbine.kernel.wave as tkw
from shark_turbine.kernel.lang.global_symbols import *
import iree.turbine.kernel as tk
import iree.turbine.kernel.lang as tkl
import iree.turbine.kernel.wave as tkw
from iree.turbine.kernel.lang.global_symbols import *
import torch

@dataclass
Expand Down

0 comments on commit bf37c5c

Please sign in to comment.