array.hpp

Go to the documentation of this file.

#pragma once
#ifndef MULTI_WRAPPER_IMPLS_ARRAY_HPP_
#define MULTI_WRAPPER_IMPLS_ARRAY_HPP_

#include "../array.hpp"
#include "../device.hpp"
#include "../event.hpp"
#include "../primary_context.hpp"
#include "../current_context.hpp"
#include "../current_device.hpp"
#include "../texture_view.hpp"

namespace cuda {

namespace array {

template <typename T, dimensionality_t NumDimensions>
array_t<T,NumDimensions> create(
    const context_t&             context,
    dimensions_t<NumDimensions>  dimensions)
{
    handle_t handle = detail_::create<T, NumDimensions>(context.handle(), dimensions);
    return wrap<T, NumDimensions>(context.device_id(), context.handle(), handle, dimensions);
}

template <typename T, dimensionality_t NumDimensions>
array_t<T,NumDimensions> create(
    const device_t&              device,
    dimensions_t<NumDimensions>  dimensions)
{
    auto pc = device.primary_context(do_not_hold_primary_context_refcount_unit);
    return create<T, NumDimensions>(pc, dimensions);
        // Note that we have no guarantee that the device's primary context
        // will continue to exist/be active when returning from this call;
        // that's the caller's responsibility
}

} // namespace array

inline context_t texture_view::context() const
{
    return context::wrap(device_id_, context_handle_);
}

inline device_t texture_view::device() const
{
    return device::get(device_id_);
}

template <typename T, dimensionality_t NumDimensions>
device_t array_t<T, NumDimensions>::device() const noexcept
{
    return device::get(device_id_);
}

template <typename T, dimensionality_t NumDimensions>
context_t array_t<T, NumDimensions>::context() const
{
    // TODO: Save the device id in the array_t as well.
    return context::wrap(device_id_, context_handle_);
}

} // namespace cuda

#endif // MULTI_WRAPPER_IMPLS_ARRAY_HPP_