File tree Expand file tree Collapse file tree 4 files changed +10
-8
lines changed Expand file tree Collapse file tree 4 files changed +10
-8
lines changed Original file line number Diff line number Diff line change 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
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 }}
Original file line number Diff line number Diff line change 11#! /bin/bash
22
3- n_ranks=4
3+ n_ranks=12
44
55if [ " $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
99fi
1010
1111if [" $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
1313else
1414 ./mfc.sh bench --mem 1 -j $( nproc) -o " $job_slug .yaml" -- -c phoenix $device_opts -n $n_ranks
1515fi
Original file line number Diff line number Diff 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.
Original file line number Diff line number Diff 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+
You can’t perform that action at this time.
0 commit comments