# Copyright 2013-2023 Lawrence Livermore National Security, LLC and other
# Spack Project Developers. See the top-level COPYRIGHT file for details.
#
# SPDX-License-Identifier: (Apache-2.0 OR MIT)

from spack.package import *


class Gdrcopy(MakefilePackage, CudaPackage):
    """A fast GPU memory copy library based on NVIDIA GPUDirect
    RDMA technology."""

    homepage = "https://github.com/NVIDIA/gdrcopy"
    url = "https://github.com/NVIDIA/gdrcopy/archive/v2.1.tar.gz"
    git = "https://github.com/NVIDIA/gdrcopy"
    maintainers("scothalverson")

    version("master", branch="master")
    version("2.3", sha256="b85d15901889aa42de6c4a9233792af40dd94543e82abe0439e544c87fd79475")
    version("2.2", sha256="e4be119809391b18c735346d24b3b398dd9421cbff47ef12befbae40d61da45f")
    version("2.1", sha256="cecc7dcc071107f77396f5553c9109790b6d2298ae29eb2dbbdd52b2a213e4ea")
    version("2.0", sha256="98320e6e980a7134ebc4eedd6cf23647104f2b3c557f2eaf0d31a02609f5f2b0")
    version("1.3", sha256="f11cdfe389b685f6636b80b4a3312dc014a385ad7220179c1318c60e2e28af3a")

    # Don't call ldconfig: https://github.com/NVIDIA/gdrcopy/pull/229
    patch("ldconfig.patch", when="@2.0:")

    depends_on("check")
    requires("+cuda")

    def build(self, spec, prefix):
        make("lib")
        make("exes")

    def install(self, spec, prefix):
        mkdir(prefix.include)
        mkdir(prefix.lib64)
        if spec.satisfies("@2.2:"):
            make("lib_install", "prefix={0}".format(self.prefix))
            make("exes_install", "prefix={0}".format(self.prefix))
        else:
            make("lib_install", "PREFIX={0}".format(self.prefix))
            make("exes_install", "PREFIX={0}".format(self.prefix))