Skip to content

Commit 3a9ac09

Browse files
authored
Remove nvidia-mathdx pip install from TE build (#1766)
Due to NVIDIA/TransformerEngine#2295
1 parent 348f515 commit 3a9ac09

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

.github/container/build-te.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,7 @@ with open("pyproject.toml", "rb") as ifile:
150150
subprocess.run(
151151
[sys.executable, "-m", "pip", "install"]
152152
+ [r for r in data["build-system"]["requires"]
153-
if r.startswith("nvidia-mathdx") or r.startswith("pybind11")])
153+
if r.startswith("pybind11")])
154154
EOF
155155

156156
# The wheel filename includes the TE commit; if this has changed since the last

0 commit comments

Comments
 (0)