ThreadwiseTensorSliceSet_v1< Data, Desc, SliceLengths, type > Struct Template Reference

ThreadwiseTensorSliceSet_v1&lt; Data, Desc, SliceLengths, type &gt; Struct Template Reference#

Composable Kernel: ck::ThreadwiseTensorSliceSet_v1< Data, Desc, SliceLengths, type > Struct Template Reference
ck::ThreadwiseTensorSliceSet_v1< Data, Desc, SliceLengths, type > Struct Template Reference

#include <threadwise_tensor_slice_set.hpp>

Public Types

using Index = MultiIndex< nDim >
 

Public Member Functions

template<typename OriginIdx , typename Buffer >
__device__ void Run (const Desc &, const OriginIdx &, Buffer &buf, const Data &initial_value) const
 

Static Public Attributes

static constexpr index_t nDim = SliceLengths::Size()
 

Member Typedef Documentation

◆ Index

template<typename Data , typename Desc , typename SliceLengths , typename enable_if< Desc::IsKnownAtCompileTime(), bool >::type = false>
using ck::ThreadwiseTensorSliceSet_v1< Data, Desc, SliceLengths, type >::Index = MultiIndex<nDim>

Member Function Documentation

◆ Run()

template<typename Data , typename Desc , typename SliceLengths , typename enable_if< Desc::IsKnownAtCompileTime(), bool >::type = false>
template<typename OriginIdx , typename Buffer >
__device__ void ck::ThreadwiseTensorSliceSet_v1< Data, Desc, SliceLengths, type >::Run ( const Desc &  ,
const OriginIdx &  ,
Buffer &  buf,
const Data &  initial_value 
) const
inline

Member Data Documentation

◆ nDim

template<typename Data , typename Desc , typename SliceLengths , typename enable_if< Desc::IsKnownAtCompileTime(), bool >::type = false>
constexpr index_t ck::ThreadwiseTensorSliceSet_v1< Data, Desc, SliceLengths, type >::nDim = SliceLengths::Size()
staticconstexpr

The documentation for this struct was generated from the following file:
  • /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck/tensor_operation/gpu/thread/threadwise_tensor_slice_set.hpp