Merge pull request cupy#8966 from kmaehashi/increase-memory-allocation

Increase host memory in Windows CI, free GPU memory in example code
chainer-ci · Feb 18, 2025 · 2be5fa7 · 2be5fa7
1 parent 88a2ee6
commit 2be5fa7
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 8 deletions.
diff --git a/.pfnci/config.pbtxt b/.pfnci/config.pbtxt
@@ -891,7 +891,7 @@ configs {
   value {
     requirement {
       cpu: 8
-      memory: 24
+      memory: 36
       disk: 10
       gpu: 2
       image: "windows"
@@ -912,7 +912,7 @@ configs {
   value {
     requirement {
       cpu: 8
-      memory: 24
+      memory: 36
       disk: 10
       gpu: 2
       image: "windows"
@@ -933,7 +933,7 @@ configs {
   value {
     requirement {
       cpu: 8
-      memory: 24
+      memory: 36
       disk: 10
       gpu: 2
       image: "windows"
@@ -954,7 +954,7 @@ configs {
   value {
     requirement {
       cpu: 8
-      memory: 24
+      memory: 36
       disk: 10
       gpu: 2
       image: "windows"
@@ -975,7 +975,7 @@ configs {
   value {
     requirement {
       cpu: 8
-      memory: 24
+      memory: 36
       disk: 10
       gpu: 2
       image: "windows"
@@ -996,7 +996,7 @@ configs {
   value {
     requirement {
       cpu: 8
-      memory: 24
+      memory: 36
       disk: 10
       gpu: 2
       image: "windows"
@@ -1017,7 +1017,7 @@ configs {
   value {
     requirement {
       cpu: 8
-      memory: 24
+      memory: 36
       disk: 10
       gpu: 2
       image: "windows"
@@ -1038,7 +1038,7 @@ configs {
   value {
     requirement {
       cpu: 8
-      memory: 24
+      memory: 36
       disk: 10
       gpu: 2
       image: "windows"

diff --git a/tests/example_tests/example_test.py b/tests/example_tests/example_test.py
@@ -2,8 +2,16 @@
 import subprocess
 import sys
 
+import cupy
+
 
 def run_example(path, *args):
+    # Free memory occupied in the main process before launching an example.
+    mempool = cupy.get_default_memory_pool()
+    pinned_mempool = cupy.get_default_pinned_memory_pool()
+    mempool.free_all_blocks()
+    pinned_mempool.free_all_blocks()
+
     examples_path = os.path.join(
         os.path.dirname(__file__), '..', '..', 'examples')
     fullpath = os.path.join(examples_path, path)