Uploaded image for project: 'Hadoop YARN'
  1. Hadoop YARN
  2. YARN-9099

GpuResourceAllocator#getReleasingGpus calculates number of GPUs in a wrong way

    XMLWordPrintableJSON

Details

    • Bug
    • Status: Resolved
    • Major
    • Resolution: Fixed
    • None
    • 3.3.0, 3.2.1, 3.1.3
    • None
    • None
    • Reviewed

    Description

      getReleasingGpus plays an important role in the calculation which happens when GpuAllocator assign GPUs to a container, see: GpuResourceAllocator#internalAssignGpus.

      If multiple GPUs are assigned to the same container, getReleasingGpus will return an invalid number.
      The iterator goes over on mappings of (GPU device, container ID) and it retrieves the container by its ID the number of times the container ID is mapped to any device.
      Then for every container, the resource value for the GPU resource is added to a running sum.
      Obviously, if a container is mapped to 2 or more devices, then the container's GPU resource counter is added to the running sum as many times as the number of GPU devices the container has.

      Example:
      Let's suppose usedDevices contains these mappings:

      • (GPU1, container1)
      • (GPU2, container1)
      • (GPU3, container2)

      GPU resource value is 2 for container1 and
      GPU resource value is 1 for container2.
      Then, if container1 is in a running state, getReleasingGpus will return 4 instead of 2.

      Attachments

        1. YARN-9099.001.patch
          2 kB
          Szilard Nemeth
        2. YARN-9099.002.patch
          1 kB
          Szilard Nemeth

        Activity

          People

            snemeth Szilard Nemeth
            snemeth Szilard Nemeth
            Votes:
            0 Vote for this issue
            Watchers:
            6 Start watching this issue

            Dates

              Created:
              Updated:
              Resolved: