Skip to content

Commit 94f3042

Browse files
pravg-amdclaude
andauthored
Pin wave to commit 775987d to fix gpt-oss export issue (#2875)
Pins wave-lang to commit 775987d38c7f7e63555fdecf000272edaf862a6a in setenv.sh for both --nightly and --nightly-cpu modes to resolve the IndexError "No current context for OpDispatcher" that occurs during torch.export for gpt-oss-20b-bfp16 model testing. --------- Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
1 parent 4685435 commit 94f3042

1 file changed

Lines changed: 14 additions & 2 deletions

File tree

scripts/setenv.sh

Lines changed: 14 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,13 @@ if [[ $BUILD_TYPE = "nightly" ]]; then
7575
pip install dataclasses-json
7676
pip install -f https://iree.dev/pip-release-links.html --upgrade --pre iree-base-compiler iree-base-runtime iree-turbine
7777
pip uninstall --y wave-lang
78-
pip install -f https://github.com/iree-org/wave/releases/expanded_assets/dev-wheels wave-lang --no-index
78+
# Install Rust (required to build wave from source)
79+
curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
80+
. "$HOME/.cargo/env"
81+
# Pinned to commit 775987d38c7f7e63555fdecf000272edaf862a6a to fix gpt-oss IndexError:
82+
# "No current context for <class 'wave_lang.kernel.ops.base.OpDispatcher'>" during torch.export
83+
# pip install -f https://github.com/iree-org/wave/releases/expanded_assets/dev-wheels wave-lang --no-index
84+
pip install git+https://github.com/iree-org/wave.git@775987d38c7f7e63555fdecf000272edaf862a6a
7985
pip install scikit-image
8086

8187
elif [[ $BUILD_TYPE = "stable" ]]; then
@@ -89,7 +95,13 @@ elif [[ $BUILD_TYPE = "nightly-cpu" ]]; then
8995
pip install shortfin[apps] -f https://github.com/nod-ai/amd-shark-ai/releases/expanded_assets/dev-wheels --pre
9096
pip install -f https://iree.dev/pip-release-links.html --upgrade --pre iree-base-compiler iree-base-runtime iree-turbine
9197
pip uninstall --y wave-lang
92-
pip install -f https://github.com/iree-org/wave/releases/expanded_assets/dev-wheels wave-lang --no-index
98+
# Install Rust (required to build wave from source)
99+
curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
100+
. "$HOME/.cargo/env"
101+
# Pinned to commit 775987d38c7f7e63555fdecf000272edaf862a6a to fix gpt-oss IndexError:
102+
# "No current context for <class 'wave_lang.kernel.ops.base.OpDispatcher'>" during torch.export
103+
# pip install -f https://github.com/iree-org/wave/releases/expanded_assets/dev-wheels wave-lang --no-index
104+
pip install git+https://github.com/iree-org/wave.git@775987d38c7f7e63555fdecf000272edaf862a6a
93105
pip install scikit-image
94106

95107
elif [[ $BUILD_TYPE = "source" ]]; then

0 commit comments

Comments
 (0)