Fix shared memory in unit test (#3204)

a5dc230a · Jinjing Zhou · GitHub · 6f93c6aa · a5dc230a · a5dc230a
Unverified Commit a5dc230a authored Jul 30, 2021 by Jinjing Zhou Committed by GitHub Jul 30, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 15 additions and 3 deletions

docker/pods/ci-cpu.yaml docker/pods/ci-cpu.yaml +8 -3

docker/pods/ci-gpu.yaml docker/pods/ci-gpu.yaml +7 -0

No files found.
--- a/docker/pods/ci-cpu.yaml
+++ b/docker/pods/ci-cpu.yaml
@@ -12,9 +12,14 @@ spec:
      requests:
        cpu: 8
    volumeMounts:
-        - name: persistent-storage
-          mountPath: /tmp/dataset
+      - name: persistent-storage
+        mountPath: /tmp/dataset
+      - name: dshm
+        mountPath: /dev/shm    
  volumes:
  - name: persistent-storage
    persistentVolumeClaim:
-      claimName: ogb-efs-claim
\ No newline at end of file
+      claimName: ogb-efs-claim
+  - name: dshm
+    emptyDir:
+      medium: Memory
--- a/docker/pods/ci-gpu.yaml
+++ b/docker/pods/ci-gpu.yaml
@@ -11,6 +11,13 @@ spec:
      resources:
        limits:
          nvidia.com/gpu: 1 # requesting 1 GPU
+      volumeMounts:
+        - name: dshm
+          mountPath: /dev/shm
+  volumes:
+  - name: dshm
+    emptyDir:
+      medium: Memory
  affinity:
    nodeAffinity:
      requiredDuringSchedulingIgnoredDuringExecution: