Set fbmem in CI runner (and use it for manual tests)

Legate allocates too much fbmem by default now for us. Not sure when/why that changed. Signed-off-by: Sebastian Berg <[email protected]>
rapidsai · Jan 9, 2025 · e333afb · e333afb
1 parent 1df1254
commit e333afb
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 14 deletions.
diff --git a/ci/manual-ci-test.py b/ci/manual-ci-test.py
@@ -1,6 +1,6 @@
 #!/usr/bin/env python
 
-# Copyright 2024 NVIDIA Corporation
+# Copyright (c) 2024-2025, NVIDIA CORPORATION
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -30,25 +30,15 @@
         "--no-capture-output",
         "-n",
         "legate-dev",
-        # TODO: Should not need legate startup,
-        # see https://github.com/nv-legate/legate.core.internal/issues/1304
-        "legate",
-        "--gpus=2",
-        "cpp/build/gtests/cpp_tests",
+        "ci/run_ctests.sh",
     ],
     "py": [
         "conda",
         "run",
         "--no-capture-output",
         "-n",
         "legate-dev",
-        # TODO: Should not need legate startup,
-        # see https://github.com/nv-legate/legate.core.internal/issues/1304
-        "legate",
-        "--gpus=2",
-        "--module",
-        "pytest",
-        "python/tests",
+        "ci/run_pytests.sh",
     ],
 }
 

diff --git a/ci/run_pytests.sh b/ci/run_pytests.sh
@@ -1,5 +1,5 @@
 #!/bin/bash
-# Copyright (c) 2024, NVIDIA CORPORATION.
+# Copyright (c) 2024-2025, NVIDIA CORPORATION.
 
 # [description]
 #
@@ -26,6 +26,7 @@ cd "$(dirname "$(realpath "${BASH_SOURCE[0]}")")"/../python/tests/
 LEGATE_TEST=${LEGATE_TEST:-1} \
 legate \
     --gpus "$(nvidia-smi -L | wc -l)"\
+    --fbmem=4000 \
     --module pytest \
     . \
     -sv \