Skip to content

GitLab

Unverified Commit a5dc230a authored by Jinjing Zhou's avatar Jinjing Zhou Committed by GitHub
Browse files

Fix shared memory in unit test (#3204)

parent 6f93c6aa
Show whitespace changes
Inline Side-by-side
Showing with 15 additions and 3 deletions
docker/pods/ci-cpu.yaml
View file @ a5dc230a
...@@ -14,7 +14,12 @@ spec: ...@@ -14,7 +14,12 @@ spec:
volumeMounts: volumeMounts:
- name: persistent-storage - name: persistent-storage
mountPath: /tmp/dataset mountPath: /tmp/dataset
- name: dshm
mountPath: /dev/shm
volumes: volumes:
- name: persistent-storage - name: persistent-storage
persistentVolumeClaim: persistentVolumeClaim:
claimName: ogb-efs-claim claimName: ogb-efs-claim
- name: dshm
emptyDir:
medium: Memory
docker/pods/ci-gpu.yaml
View file @ a5dc230a
...@@ -11,6 +11,13 @@ spec: ...@@ -11,6 +11,13 @@ spec:
resources: resources:
limits: limits:
nvidia.com/gpu: 1 # requesting 1 GPU nvidia.com/gpu: 1 # requesting 1 GPU
volumeMounts:
- name: dshm
mountPath: /dev/shm
volumes:
- name: dshm
emptyDir:
medium: Memory
affinity: affinity:
nodeAffinity: nodeAffinity:
requiredDuringSchedulingIgnoredDuringExecution: requiredDuringSchedulingIgnoredDuringExecution:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment