Make Ray work on GH200 (#40816)

Signed-off-by: Philipp Moritz <[email protected]>
ray-project · Nov 17, 2023 · 0e2a523 · 0e2a523
1 parent 5df518a
commit 0e2a523
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/bazel/BUILD.jemalloc b/bazel/BUILD.jemalloc
@@ -13,7 +13,13 @@ configure_make(
     copts = ["-fPIC"],
     args = ["-j"],
     out_shared_libs = ["libjemalloc.so"],
-    configure_options = ["--disable-static", "--enable-prof"],
+    # See https://salsa.debian.org/debian/jemalloc/-/blob/c0a88c37a551be7d12e4863435365c9a6a51525f/debian/rules#L8-23
+    # for why we are setting "--with-lg-page" on non x86 hardware here.
+    configure_options = ["--disable-static", "--enable-prof"] +
+        select({
+            "@platforms//cpu:x86_64": [],
+            "//conditions:default": ["--with-lg-page=16"],
+        }),
     visibility = ["//visibility:public"],
 )