Add nanobind to mlir-nvidia Dockerfile (llvm#327)

luporl · Dec 5, 2024 · 7424d71 · 7424d71
1 parent 0f057bd
commit 7424d71
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/buildbot/mlir-nvidia/Dockerfile b/buildbot/mlir-nvidia/Dockerfile
@@ -48,10 +48,6 @@ RUN apt-get update \
 # https://github.com/pypa/manylinux
 RUN python3 -m pip install --upgrade pip==23.1.2
 
-# Refer to mlir/lib/Bindings/Python/requirements.txt. Listed explicitly here
-# and version pinned for consistency as this is a bot.
-RUN python3 -m pip install numpy==1.24.3 pybind11==2.10.4 PyYAML==5.3.1
-
 # Ubuntu ships with old cmake version, install the latest one
 # from https://apt.kitware.com/
 RUN wget -O - https://apt.kitware.com/keys/kitware-archive-latest.asc 2>/dev/null \

diff --git a/buildbot/osuosl/master/config/builders.py b/buildbot/osuosl/master/config/builders.py
@@ -1475,6 +1475,7 @@
                     llvm_srcdir="llvm.src",
                     obj_dir="llvm.obj",
                     clean=True,
+                    install_pip_requirements=True,
                     targets = ['check-mlir-build-only'],
                     checks = ['check-mlir'],
                     depends_on_projects=['llvm','mlir'],
@@ -1500,6 +1501,7 @@
                     llvm_srcdir="llvm.src",
                     obj_dir="llvm.obj",
                     clean=True,
+                    install_pip_requirements=True,
                     targets = ['check-mlir-build-only'],
                     checks = ['check-mlir'],
                     depends_on_projects=['llvm','mlir'],