cuda::experimental::stf::slice_graph_interface

template<typename T, size_t dimensions = 1>
class slice_graph_interface : public cuda::experimental::stf::graph_data_interface<slice<T, 1>>

Slice data interface implementation over CUDA graphs.

The type manipulated will be slice<T, dimensions>.

Template Parameters:

• T – Element type for the slice

• dimensions – rank of the slice

Public Types

using base = graph_data_interface<slice<T, dimensions>>

Alias for the base class.

using mutable_value_type = typename ::std::remove_const<T>::type

using shape_t = typename base::shape_t

using element_type = T

Public Functions

inline slice_graph_interface(slice<T, dimensions> s)

Constructor from slice.

Parameters:

s – either a slice or the shape of a slice

inline slice_graph_interface(shape_of<slice<T, dimensions>> s)

Constructor from shape.

Parameters:

s – either a slice or the shape of a slice

inline virtual void data_allocate(

backend_ctx_untyped &bctx,

block_allocator_untyped &custom_allocator,

const data_place &memory_node,

instance_id_t instance_id,

::std::ptrdiff_t &s,

void **extra_args,

event_list &prereqs,

) override

Implementation of interface primitive.

inline virtual void data_deallocate(

backend_ctx_untyped &bctx,

block_allocator_untyped &custom_allocator,

const data_place &memory_node,

instance_id_t instance_id,

void *extra_args,

event_list &prereqs,

) final

Implementation of interface primitive.

inline virtual cudaGraphNode_t graph_data_copy(

cudaMemcpyKind kind,

instance_id_t src_instance_id,

instance_id_t dst_instance_id,

cudaGraph_t graph,

const cudaGraphNode_t *input_nodes,

size_t input_cnt,

) override

Implementation of interface primitive.

inline virtual bool pin_host_memory(

instance_id_t instance_id,

) override

Implementation of interface primitive.

inline virtual void unpin_host_memory(instance_id_t) override

Implementation of interface primitive.

inline virtual void data_copy(

backend_ctx_untyped &ctx_,

const data_place &dst_memory_node,

instance_id_t dst_instance_id,

const data_place &src_memory_node,

instance_id_t src_instance_id,

event_list &prereqs,

) override

Copy data from one memory node to another, returning a prerequisite event list.

Parameters:

• ctx – Backend context state

• dst_memory_node – The destination memory node

• dst_instance_id – The destination instance ID

• src_memory_node – The source memory node

• src_instance_id – The source instance ID

• arg – Additional arguments required for copying data

• prereqs – Prerequisite event list, will be updated as a side effect

template<typename T>
inline const T &shape() const

Get the part of the data interface that is common to all data instances.

Template Parameters:

T – The type of the data shape

Returns:

A const reference to the data shape

inline T &instance(instance_id_t instance_id)

inline const T &instance(instance_id_t instance_id) const

inline virtual size_t data_hash(

instance_id_t instance_id,

) const final

Get the hash of the data representation for the given instance ID.

Parameters:

instance_id – The ID of the data instance

Returns:

The hash of the data representation

inline size_t data_footprint() const final

inline ::std::optional<cudaMemoryType> get_memory_type(

instance_id_t,

)

template<typename T>
inline const T &instance_const(

instance_id_t instance_id,

) const

Get a const reference to the data instance of the given ID.

Template Parameters:

T – The type of the data instance

Parameters:

instance_id – The ID of the data instance

Returns:

A const reference to the data instance

template<typename backend_ctx_untyped>
inline instance_id_t get_default_instance_id(

backend_ctx_untyped&,

const logical_data_untyped &d,

task &tp,

) const

Returns the index (ID) of the data instance for this logical data in the current task context.

If this is called outside a task, this will result in an error.

Parameters:

• ctx – The backend context state

• d – The logical data_untyped

Returns:

The ID of the data instance for this logical data

inline virtual bool is_void_interface() const

Indicates whether this is a void data interface, which permits to skip some operations to allocate or move data for example.