Config for mlir-nvidia buildbot.
Configuration for new buildbot.
Patch by Christian Kühnel <kuhnel@google.com>
Differential Revision: https://reviews.llvm.org/D74200
diff --git a/buildbot/osuosl/master/config/builders.py b/buildbot/osuosl/master/config/builders.py
index b31f82e..8d94b6e 100644
--- a/buildbot/osuosl/master/config/builders.py
+++ b/buildbot/osuosl/master/config/builders.py
@@ -1018,6 +1018,33 @@
]
+# Builders for MLIR
+def _get_mlir_builders():
+ return [
+ {'name': "mlir-nvidia",
+ 'mergeRequests': False,
+ 'slavenames':["mlir-nvidia"],
+ 'builddir':"mlir-nvidia",
+ 'factory': UnifiedTreeBuilder.getCmakeWithNinjaBuildFactory(
+ llvm_srcdir="llvm.src",
+ obj_dir="llvm.obj",
+ clean=True,
+ depends_on_projects=['llvm','mlir'],
+ extra_configure_args=[
+ '-DLLVM_BUILD_EXAMPLES=ON'
+ '-DLLVM_ENABLE_CXX1Y=Y '
+ '-DLLVM_TARGETS_TO_BUILD="host;NVPTX"'
+ '-DLLVM_ENABLE_PROJECTS=mlir'
+ '-DMLIR_CUDA_RUNNER_ENABLED=1'
+ '-DCMAKE_CUDA_COMPILER=/usr/local/cuda/bin/nvcc'
+ ],
+ env={
+ 'CC':'clang',
+ 'CXX': 'clang++',
+ 'LD': 'lld',
+ })},
+ ]
+
# Sanitizer builders.
def _get_sanitizer_builders():
return [
@@ -1513,6 +1540,10 @@
b['category'] = 'lldb'
yield b
+ for b in _get_mlir_builders():
+ b['category'] = 'mlir'
+ yield b
+
for b in _get_sanitizer_builders():
b['category'] = 'sanitizer'
yield b
diff --git a/buildbot/osuosl/master/config/slaves.py b/buildbot/osuosl/master/config/slaves.py
index d88f626..769ac4f 100644
--- a/buildbot/osuosl/master/config/slaves.py
+++ b/buildbot/osuosl/master/config/slaves.py
@@ -215,4 +215,7 @@
# CentOS 7.5.1804 on Intel(R) Xeon(R) Gold 6126 CPU @ 2.60GHz, 96GB RAM
create_slave("nec-arrproto41", properties={'jobs': 12}, max_builds=1),
+
+ # Uubntu 18.04 amd64 on Google Cloud, 16 core, Nvidia Tesla T4
+ create_slave("mlir-nvidia", properties={'jobs': 16}, max_builds=1),
]
