Skip to content

Commit 0db6323

Browse files
authored
Add reframe.yaml config for pytorch uenv (#298)
1 parent 4f5e642 commit 0db6323

File tree

2 files changed

+17
-0
lines changed

2 files changed

+17
-0
lines changed

recipes/pytorch/v2.9.1/gh200/environments.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,7 @@ pytorch-env:
7272
- py-onnxscript
7373
- py-regex
7474
- py-tensorboard
75+
- py-psutil
7576
#- py-vllm
7677
- dynolog +use-prometheus +with-unwind +with-gflags
7778
# packages which don't compile with +ipo
Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
1+
default:
2+
features:
3+
- cuda
4+
- mpi
5+
- cray-mpich
6+
- nccl
7+
- nccl-tests
8+
- openmp
9+
- osu-micro-benchmarks
10+
- serial
11+
- pytorch
12+
cc: mpicc
13+
cxx: mpic++
14+
ftn: mpifort
15+
views:
16+
- default

0 commit comments

Comments
 (0)