Skip to content

New API call 'setup_gpu'

Andreas Marek requested to merge gpu_nccl into master_pre_stage

New API call to setup the gpu usage, speeds up the setup tremendously

  • must be called after the call to setup
  • must be called after setting one of the keywords 'nvidia-gpu', 'amd-gpu' or 'intel-gpu'

Merge request reports