Skip to content

Commit f9d1e35

Browse files
Archith IyerArchith Iyer
authored andcommitted
fix
1 parent 973b084 commit f9d1e35

File tree

4 files changed

+10
-8
lines changed

4 files changed

+10
-8
lines changed

.github/workflows/bench.yml

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,8 @@ jobs:
2020

2121
self:
2222
name: Georgia Tech | Phoenix (NVHPC)
23-
if: github.repository == 'MFlowCode/MFC'
23+
if: github.repository == 'MFlowCode/MFC' && needs.file-changes.outputs.checkall == 'true'
24+
needs: file-changes
2425
strategy:
2526
matrix:
2627
device: ['cpu', 'gpu']
@@ -33,12 +34,12 @@ jobs:
3334
ACTIONS_ALLOW_USE_UNSECURE_NODE_VERSION: true
3435
steps:
3536
- name: Clone - PR
36-
uses: actions/checkout@v3
37+
uses: actions/checkout@v4
3738
with:
3839
path: pr
3940

4041
- name: Clone - Master
41-
uses: actions/checkout@v3
42+
uses: actions/checkout@v4
4243
with:
4344
repository: MFlowCode/MFC
4445
ref: master
@@ -56,7 +57,7 @@ jobs:
5657
(cd pr && ./mfc.sh bench_diff ../master/bench-${{ matrix.device }}.yaml ../pr/bench-${{ matrix.device }}.yaml)
5758
5859
- name: Archive Logs
59-
uses: actions/upload-artifact@v3
60+
uses: actions/upload-artifact@v4
6061
if: always()
6162
with:
6263
name: logs-${{ matrix.device }}

.github/workflows/phoenix/bench.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
#!/bin/bash
22

3-
n_ranks=4
3+
n_ranks=12
44

55
if [ "$job_device" == "gpu" ]; then
66
n_ranks=$(nvidia-smi -L | wc -l) # number of GPUs on node
@@ -9,7 +9,7 @@ if [ "$job_device" == "gpu" ]; then
99
fi
1010

1111
if ["$job_device" == "gpu"]; then
12-
./mfc.sh bench --mem 8 -j $(nproc) -o "$job_slug.yaml" -- -c phoenix $device_opts -n $n_ranks
12+
./mfc.sh bench --mem 12 -j $(nproc) -o "$job_slug.yaml" -- -c phoenix $device_opts -n $n_ranks
1313
else
1414
./mfc.sh bench --mem 1 -j $(nproc) -o "$job_slug.yaml" -- -c phoenix $device_opts -n $n_ranks
1515
fi

.github/workflows/phoenix/submit.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ sbatch <<EOT
4141
#SBATCH --account=gts-sbryngelson3 # charge account
4242
#SBATCH -N1 # Number of nodes required
4343
$sbatch_device_opts
44-
#SBATCH -t 04:00:00 # Duration of the job (Ex: 15 mins)
44+
#SBATCH -t 02:00:00 # Duration of the job (Ex: 15 mins)
4545
#SBATCH -q embers # QOS Name
4646
#SBATCH -o$job_slug.out # Combined output and error messages file
4747
#SBATCH -W # Do not exit until the submitted job terminates.

toolchain/mfc/bench.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ def bench(targets = None):
6161
["--targets"] + [t.name for t in targets] +
6262
["--output-summary", summary_filepath] +
6363
case.args +
64-
["--", ARG('mem')],
64+
["--", "--gbpp", ARG('mem')],
6565
stdout=log_file,
6666
stderr=subprocess.STDOUT)
6767

@@ -168,3 +168,4 @@ def _lock_to_str(lock):
168168

169169
if err != 0:
170170
raise MFCException("Benchmarking failed")
171+

0 commit comments

Comments
 (0)