Skip to content

Commit 43104e7

Browse files
committed
FST: allow choosing the GPU backend of the WITHGPU stage
The stage run with ENABLE_GPU_TEST=1 was hardcoded to CUDA. FST_GPUTYPE selects it, e.g. FST_GPUTYPE=METAL on macOS; the default stays CUDA.
1 parent 3db945b commit 43104e7

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

‎prodtests/full_system_test.sh‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,7 @@ NJOBS=${NJOBS:-"${NCPUS}"}
4848
SHMSIZE=${SHMSIZE:-8000000000} # Size of shared memory for messages (use 128 GB for 550 event full TF)
4949
#TPCTRACKERSCRATCHMEMORY=${TPCTRACKERSCRATCHMEMORY:-8000000000} # Not needed by default
5050
ENABLE_GPU_TEST=${ENABLE_GPU_TEST:-0} # Run the full system test also on the GPU
51+
FST_GPUTYPE=${FST_GPUTYPE:-CUDA} # GPU backend used in case ENABLE_GPU_TEST=1 (CUDA / HIP / METAL)
5152
GPUMEMSIZE=${GPUMEMSIZE:-6000000000} # Size of GPU memory to use in case ENABBLE_GPU_TEST=1
5253
NTIMEFRAMES=${NTIMEFRAMES:-1} # Number of time frames to process
5354
TFDELAY=${TFDELAY:-100} # Delay in seconds between publishing time frames
@@ -281,7 +282,7 @@ for STAGE in $STAGES; do
281282
export WORKFLOW_PARAMETERS=
282283
if [[ "$STAGE" = "WITHGPU" ]]; then
283284
export CREATECTFDICT=0
284-
export GPUTYPE=CUDA
285+
export GPUTYPE=$FST_GPUTYPE
285286
export HOSTMEMSIZE=1000000000
286287
export SYNCMODE=1
287288
export CTFINPUT=0

0 commit comments

Comments
 (0)