Skip to content
Snippets Groups Projects
Select Git revision
  • 4a053716580ed55017388072200e9437546a887a
  • main default protected
2 results

mysbatch.sh

Blame
    • mysbatch.sh 996 B
    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    29
    30
    31
    32
    33
    34
    35
    36
    37
    38
    39
    40
    41
    42
    43
    44
    45
    46
    47
    48
    49
    50
    51
    52
    53
    54
    55
    56
    57
    58 
    

    

    

    #!/bin/bash -l
# Standard output and error:
##SBATCH -o ./job.out.%j
##SBATCH -e ./job.err.%j
# Initial working directory:
#SBATCH -D ./
# Job name
#SBATCH -J test_scalapack_gpu

#SBATCH --nodes=2
#SBATCH --ntasks-per-node=4
##SBATCH --ntasks=1
#SBATCH --cpus-per-task=1

#SBATCH --constraint="gpu"

# cobra/talos
#SBATCH --gres=gpu:v100:2
##SBATCH --mem=92500 

# raven
##SBATCH --gres=gpu:a100:4
##SBATCH --mem=125000 
##SBATCH --nvmps # for efficient concurrent access to GPUs from multiple processes: only raven

#SBATCH --mail-type=all
##SBATCH --mail-user=userid@example.mpg.de
#SBATCH --time=00:10:00

# verbose
set -v

export BLOCK_SIZE_PILAENV=5120

k=4
N0=10240
N=$((k*N0))
NB=1024

postfix="_scalapack_wrapper_gpu"

#____________________________________________________________

FilenameC="main"
FilenameExe=$FilenameC$postfix

source compile$postfix.sh

#____________________________________________________________
# run

nvidia-smi
nvcc --version

srun $FilenameExe $N $NB

exit $?