<commit_before>###############################################################################
# System
###############################################################################
# Target system to test on.
system:
  "linux":
  "windows":

###############################################################################
# OS
###############################################################################
# Name and version of the operating system to test on.
os:
  "ubuntu:18.04":
    system: "linux"
  "ubuntu:20.04":
    system: "linux"
  "centos:7":
    system: "linux"
  "centos:8":
    system: "linux"
  "ws:2016":
    system: "windows"

###############################################################################
# CUDA & ROCm
###############################################################################
# The platform to test with.
# `cuda` and `rocm` are mutually exclusive; when one is set, the other must be null.
# `full_version` is the version used in base Docker images.
# When adding a new CUDA version, you may also need to add the version to the
# list of supported CUDA versions in the optional libraries.
cuda:
  null:
    full_version: null
  "10.2":
    full_version: "10.2"
  "11.0":
    full_version: "11.0.3"
  "11.1":
    full_version: "11.1.1"
  "11.2":
    full_version: "11.2.1"
  "11.3":
    full_version: "11.3.1"
  "11.4":
    full_version: "11.4.0"
  "11.5":
    full_version: "11.5.0"
rocm:
  null:
    full_version: null
    packages: []
  "4.0":
    full_version: "4.0.1"
    packages: ['rocm-dev', 'hipblas', 'hipsparse', 'hipcub',
               'rocsparse', 'rocrand', 'rocthrust', 'rocsolver', 'rocfft',
               'rocprim', 'rccl']
  "4.2":
    full_version: "4.2"
    packages: ['rocm-dev', 'hipblas', 'hipfft', 'hipsparse', 'hipcub',
               'rocsparse', 'rocrand', 'rocthrust', 'rocsolver', 'rocfft',
               'rocprim', 'rccl']
  "4.3":
    full_version: "4.3"
    packages: ['rocm-dev', 'hipblas', 'hipfft', 'hipsparse', 'hipcub',
               'rocsparse', 'rocrand', 'rocthrust', 'rocsolver', 'rocfft',
               'rocprim', 'rccl']

###############################################################################
# CUDA Optional Libraries
###############################################################################
# The optional libraries to install: NCCL, cuTENSOR, cuSPARSELt and cuDNN.
# `spec` must be a version specifier accepted by apt and yum.
# `cuda` lists the supported CUDA version.
# `cuda.alias` (cuDNN only) is an alias name for package name suffix.
nccl:
  # https://docs.nvidia.com/deeplearning/nccl/archives/
  # https://developer.nvidia.com/nccl/nccl-download
  null:
    spec: null
  "2.8":
    spec: "2.8.*"
    cuda:
      # "10.1":
      "10.2":
      "11.0":
      "11.1":<commit_msg>bump ROCm version for CI from 4.3 to 4.3.1<commit_after>###############################################################################
# System
###############################################################################
# Target system to test on.
system:
  "linux":
  "windows":

###############################################################################
# OS
###############################################################################
# Name and version of the operating system to test on.
os:
  "ubuntu:18.04":
    system: "linux"
  "ubuntu:20.04":
    system: "linux"
  "centos:7":
    system: "linux"
  "centos:8":
    system: "linux"
  "ws:2016":
    system: "windows"

###############################################################################
# CUDA & ROCm
###############################################################################
# The platform to test with.
# `cuda` and `rocm` are mutually exclusive; when one is set, the other must be null.
# `full_version` is the version used in base Docker images.
# When adding a new CUDA version, you may also need to add the version to the
# list of supported CUDA versions in the optional libraries.
cuda:
  null:
    full_version: null
  "10.2":
    full_version: "10.2"
  "11.0":
    full_version: "11.0.3"
  "11.1":
    full_version: "11.1.1"
  "11.2":
    full_version: "11.2.1"
  "11.3":
    full_version: "11.3.1"
  "11.4":
    full_version: "11.4.0"
  "11.5":
    full_version: "11.5.0"
rocm:
  null:
    full_version: null
    packages: []
  "4.0":
    full_version: "4.0.1"
    packages: ['rocm-dev', 'hipblas', 'hipsparse', 'hipcub',
               'rocsparse', 'rocrand', 'rocthrust', 'rocsolver', 'rocfft',
               'rocprim', 'rccl']
  "4.2":
    full_version: "4.2"
    packages: ['rocm-dev', 'hipblas', 'hipfft', 'hipsparse', 'hipcub',
               'rocsparse', 'rocrand', 'rocthrust', 'rocsolver', 'rocfft',
               'rocprim', 'rccl']
  "4.3":
    full_version: "4.3.1"
    packages: ['rocm-dev', 'hipblas', 'hipfft', 'hipsparse', 'hipcub',
               'rocsparse', 'rocrand', 'rocthrust', 'rocsolver', 'rocfft',
               'rocprim', 'rccl']

###############################################################################
# CUDA Optional Libraries
###############################################################################
# The optional libraries to install: NCCL, cuTENSOR, cuSPARSELt and cuDNN.
# `spec` must be a version specifier accepted by apt and yum.
# `cuda` lists the supported CUDA version.
# `cuda.alias` (cuDNN only) is an alias name for package name suffix.
nccl:
  # https://docs.nvidia.com/deeplearning/nccl/archives/
  # https://developer.nvidia.com/nccl/nccl-download
  null:
    spec: null
  "2.8":
    spec: "2.8.*"
    cuda:
      # "10.1":
      "10.2":
      "11.0":
      "11.1":
{'fraction_non_alphanumeric': 0.35189468066180396, 'fraction_numerical': 0.03344600604874577, 'mean_word_length': 3.834049871023216, 'pattern_counts': {'":': 44, '<': 3, '<?xml version=': 0, '>': 3, 'https://': 4, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 2, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '5478552', 'n_tokens_mistral': 1907, 'n_tokens_neox': 1792, 'n_words': 457}