Skip to content
forked from kokkos/hpcbind

Binding utilities used for MPI, OpenMP and GPUs

License

Notifications You must be signed in to change notification settings

bartgol/hpcbind

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

39 Commits
 
 
 
 
 
 

Repository files navigation

Purpose

A script to set the process mask, OMP environment variables and CUDA environment variables to sane values if possible. Uses hwloc and nvidia-smi if available. Will preserve the current process binding, so it is safe to use with a queuing system or mpiexec.

Tested with hwloc versions >= 1.10

Usage: hpcbind <options> -- command ...
  Set the process mask, OMP environment variables and CUDA environment
  variables to sane values if possible. Uses hwloc and nvidia-smi if
  available.  Will preserve the current process binding, so it is safe
  to use with a queuing system or mpiexec.

Options:
  --no-hwloc-bind       Disable binding
  --proc-bind=<LOC>     Set the initial process mask for the script
                        LOC can be any valid location argument for
                        hwloc-calc  Default: all
  --whole-system        hpcbind will ignore the its parent process binding
  --distribute=N        Distribute the current cpuset into N partitions
  --distribute-partition=I
                        Use the i'th partition (zero based)
  --visible-gpus=<L>    Comma separated list of gpu ids
                        Default: CUDA_VISIBLE_DEVICES or all gpus in
                        sequential order
  --ignore-queue        Ignore queue job id when choosing visible GPU and partition
  --no-gpu-mapping      Do not set CUDA_VISIBLE_DEVICES
  --openmp=M.m          Set env variables for the given OpenMP version
                        Default: 4.0
  --openmp-ratio=N      Divide ratio of the cpuset to use for OpenMP
                        Default: 1
  --openmp-places=<Op>  Op=threads|cores|sockets. Default: threads
  --no-openmp-proc-bind Set OMP_PROC_BIND to false and unset OMP_PLACES
  --force-openmp-num-threads=N
                        Override logic for selecting OMP_NUM_THREADS
  --force-openmp-proc-bind=<OP>
                        Override logic for selecting OMP_PROC_BIND
  --no-openmp-nested    Set OMP_NESTED to false
  --output-prefix=<P>   Save the output to files of the form
                        P.hpcbind.N, P.stdout.N and P.stderr.N where P is
                        the prefix and N is the rank (no spaces)
  --output-mode=<Op>    How console output should be handled.
                        Options are all, rank0, and none.  Default: rank0
  --lstopo              Show bindings in lstopo
  -v|--verbose          Print bindings and relevant environment variables
  -h|--help             Show this message

Sample Usage:

  Split the current process cpuset into 4 and use the 3rd partition
    hpcbind --distribute=4 --distribute-partition=2 -v -- command ...

  Launch 16 jobs over 4 nodes with 4 jobs per node using only the even pus
  and save the output to rank specific files
    mpiexec -N 16 -npernode 4 hpcbind --whole-system --proc-bind=pu:even \
      --distribute=4 -v --output-prefix=output  -- command ...

  Bind the process to all even cores
    hpcbind --proc-bind=core:even -v -- command ...

  Bind the the even cores of socket 0 and the odd cores of socket 1
    hpcbind --proc-bind='socket:0.core:even socket:1.core:odd' -v -- command ...

  Skip GPU 0 when mapping visible devices
    hpcbind --distribute=4 --distribute-partition=0 --visible-gpus=1,2 -v -- command ...

  Display the current bindings
    hpcbind --proc-bind=numa:0 -- command

  Display the current bindings using lstopo
    hpcbind --proc-bind=numa:0.core:odd --lstopo

See https://github.com/open-mpi/hwloc for more information about hwloc

About

Binding utilities used for MPI, OpenMP and GPUs

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Shell 100.0%