From bf37c5c1e1a8996295e8ec19cc91035ae5019743 Mon Sep 17 00:00:00 2001
From: saienduri <77521230+saienduri@users.noreply.github.com>
Date: Sun, 6 Oct 2024 12:02:20 -0700
Subject: [PATCH] Adapt to iree-turbine naming changes. (#11)

This commit adapts to the recent changes made to the packaging naming in
the iree-org/iree-turbine repo:
https://github.com/iree-org/iree-turbine/pull/197
---
 .github/workflows/run_bench.yml | 2 +-
 README.md                       | 2 +-
 gemmbench/gemm_utils.py         | 8 ++++----
 3 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/.github/workflows/run_bench.yml b/.github/workflows/run_bench.yml
index 63bdefd..b126438 100644
--- a/.github/workflows/run_bench.yml
+++ b/.github/workflows/run_bench.yml
@@ -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: |
diff --git a/README.md b/README.md
index c5ee600..386b36e 100644
--- a/README.md
+++ b/README.md
@@ -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
diff --git a/gemmbench/gemm_utils.py b/gemmbench/gemm_utils.py
index b84bf0a..03c4228 100644
--- a/gemmbench/gemm_utils.py
+++ b/gemmbench/gemm_utils.py
@@ -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