link.hpp

Go to the documentation of this file.

#pragma once
#ifndef CUDA_API_WRAPPERS_LINK_HPP_
#define CUDA_API_WRAPPERS_LINK_HPP_

#include "current_context.hpp"
#include "link_options.hpp"

#if __cplusplus >= 201703L
#include <filesystem>
#endif

namespace cuda {

class device_t;
class module_t;
class link_t;

namespace link {

enum class input_kind_t {
    cubin,    
    ptx,      
    fatbin,   
    object,   
    library,  
};

using handle_t = CUlinkState;

inline link_t wrap(
    device::id_t device_id,
    context::handle_t context_handle,
    link::handle_t handle,
    const link::options_t &options,
    bool take_ownership = false) noexcept;

namespace input {

struct image_t : memory::region_t {
    const char *name; 
    link::input_kind_t type; 
};

struct file_t {
    const char *path;
    link::input_kind_t type; 
};

} // namespace input

namespace detail_ {

inline void destroy(handle_t handle, context::handle_t context_handle, device::id_t device_id)
{
    CAW_SET_SCOPE_CONTEXT(context_handle);
    auto status = cuLinkDestroy(handle);
    throw_if_error_lazy(status,
        ::std::string("Failed destroying the link ") + cuda::detail_::ptr_as_hex(handle)
        + " in " + context::detail_::identify(context_handle)
        + " on " + device::detail_::identify(device_id));
}

} // namespace detail_

} // namespace link

class link_t {

public: // getters
    device::id_t device_id() const noexcept
    { return device_id_; };

    context::handle_t context_handle() const noexcept
    { return context_handle_; }

    bool is_owning() const noexcept
    { return owning; }

    device_t device() const;

    context_t context() const;

public:
    memory::region_t complete() const
    {
        void *cubin_output_start;
        size_t cubin_output_size;
        auto status = cuLinkComplete(handle_, &cubin_output_start, &cubin_output_size);
        throw_if_error_lazy(status,
            "Failed completing the link with state at address " + cuda::detail_::ptr_as_hex(handle_));
        return memory::region_t{cubin_output_start, cubin_output_size};
    }

    void add(link::input::image_t image, const link::options_t &ptx_compilation_options = {}) const
    {
        auto marshalled_options = link::detail_::marshal(ptx_compilation_options);
        auto status = cuLinkAddData(
            handle_,
            static_cast<CUjitInputType>(image.type),
            image.data(), // TODO: Is this really safe?
            image.size(),
            image.name,
            marshalled_options.count(),
            const_cast<link::detail_::option_t *>(marshalled_options.options()),
            const_cast<void **>(marshalled_options.values())
        );
        throw_if_error_lazy(status,
            "Failed adding input " + ::std::string(image.name) + " of type "
            + ::std::to_string(static_cast<int>(image.type)) + " to a link.");
    }

    void add_file(link::input::file_t file_input, const link::options_t &options = {}) const
    {
        auto marshalled_options = link::detail_::marshal(options);
        auto status = cuLinkAddFile(
            handle_,
            static_cast<CUjitInputType_enum>(file_input.type),
            file_input.path,
            marshalled_options.count(),
            const_cast<link::detail_::option_t *>(marshalled_options.options()),
            const_cast<void **>(marshalled_options.values())
        );
        throw_if_error_lazy(status,
            "Failed loading an object of type " + ::std::to_string(static_cast<int>(file_input.type))
            + " from file " + file_input.path);
    }

    void add_file(
        const char* path,
        link::input_kind_t file_contents_type,
        const link::options_t &options = {}) const
    {
        auto link_file_spec = link::input::file_t { path, file_contents_type };
        return add_file(link_file_spec, options);
    }

#if __cplusplus >= 201703L
    void add_file(
        const ::std::filesystem::path& path,
        link::input_kind_t file_contents_type,
        const link::options_t &options = {}) const
    {
        return add_file(path.c_str(), file_contents_type, options);
    }
#endif

protected: // constructors

    link_t(
        device::id_t device_id,
        context::handle_t context,
        link::handle_t handle,
        const link::options_t &options,
        bool take_ownership) noexcept
        : device_id_(device_id), context_handle_(context), handle_(handle), options_(options), owning(take_ownership)
    {}

public: // friendship

    friend link_t link::wrap(device::id_t, context::handle_t, link::handle_t, const link::options_t &, bool) noexcept;

public: // constructors and destructor

    link_t(const link_t &) = delete;

    link_t(link_t &&other) noexcept:
        link_t(other.device_id_, other.context_handle_, other.handle_, other.options_, other.owning)
    {
        other.owning = false;
    };

    ~link_t() DESTRUCTOR_EXCEPTION_SPEC
    {
        if (not owning) { return; }
#ifndef THROW_IN_DESTRUCTORS
        try
#endif
        {
            link::detail_::destroy(handle_, context_handle_, device_id_);
        }
#ifndef THROW_IN_DESTRUCTORS
        catch (...) {}
#endif
    }

public: // operators

    link_t &operator=(const link_t &) = delete;

    link_t &operator=(link_t &&other) noexcept
    {
        ::std::swap(device_id_, other.device_id_);
        ::std::swap(context_handle_, other.context_handle_);
        ::std::swap(handle_, other.handle_);
        ::std::swap(options_, other.options_);
        ::std::swap(owning, owning);
        return *this;
    }

protected: // data members
    device::id_t device_id_;
    context::handle_t context_handle_;
    link::handle_t handle_;
    link::options_t options_;
    bool owning;
    // this field is mutable only for enabling move construction; other
    // than in that case it must not be altered
};

namespace link {

inline link_t create(const link::options_t &options = link::options_t{})
{
    handle_t new_link_handle;
    auto marshalled_options = link::detail_::marshal(options);
    auto status = cuLinkCreate(
        marshalled_options.count(),
        const_cast<link::detail_::option_t *>(marshalled_options.options()),
        const_cast<void **>(marshalled_options.values()),
        &new_link_handle
    );
    throw_if_error_lazy(status, "Failed creating a new link ");
    auto do_take_ownership = true;
    auto context_handle = context::current::detail_::get_handle();
    auto device_id = context::current::detail_::get_device_id();
    return wrap(
        device_id,
        context_handle,
        new_link_handle,
        options,
        do_take_ownership);
}

inline link_t wrap(
    device::id_t             device_id,
    context::handle_t        context_handle,
    link::handle_t           handle,
    const link::options_t &  options,
    bool                     take_ownership) noexcept
{
    return link_t{device_id, context_handle, handle, options, take_ownership};
}

} // namespace link

} // namespace cuda

#endif // CUDA_API_WRAPPERS_LINK_HPP_