41 typename FragmentElement_,
45 template <
int kAccessSize,
48 typename FragmentElement_,
60 static CUTLASS_DEVICE
void load(
AccessType& value, Scalar_
const* pointer,
int offset) {
61 value.load(&pointer[offset], kStride);
65 template <
int kAccessSize,
68 typename FragmentElement_,
80 static CUTLASS_DEVICE
void load(
AccessType& value, Scalar_
const* pointer,
int offset) {
89 typename FragmentElement_,
93 template <
int kAccessSize,
96 typename FragmentElement_,
108 static CUTLASS_DEVICE
void store(
AccessType const& value, Scalar_* pointer,
int offset) {
109 value.store(&pointer[offset], kStride);
113 template <
int kAccessSize,
116 typename FragmentElement_,
128 static CUTLASS_DEVICE
void store(
AccessType const& value, Scalar_* pointer,
int offset) {
Definition: fragment_load_store.h:43
Vectorize< Scalar_, kAccessSize >::Type AccessType
The input type.
Definition: fragment_load_store.h:125
FragmentElement_ AccessType
The input type.
Definition: fragment_load_store.h:105
Vectorize< Scalar_, kAccessSize >::Type AccessType
The output type.
Definition: fragment_load_store.h:77
static CUTLASS_DEVICE void load(AccessType &dst, Scalar_ const *pointer, int offset)
The load function.
Definition: load_store.h:59
static CUTLASS_DEVICE void store(AccessType const &value, Scalar_ *pointer, int offset)
The store function.
Definition: fragment_load_store.h:108
static CUTLASS_DEVICE void store(AccessType const &value, Scalar_ *pointer, int offset)
The store function.
Definition: fragment_load_store.h:128
Kind
Definition: load_store.h:40
static CUTLASS_DEVICE void store(AccessType const &src, Scalar_ *pointer, int offset)
The store function.
Definition: load_store.h:136
Kind
Definition: tile_iterator.h:67
static CUTLASS_DEVICE void load(AccessType &value, Scalar_ const *pointer, int offset)
The load function.
Definition: fragment_load_store.h:80
Defines abstractions for efficiently loading and storing vectors to memory.
Defines a 1D vector of elements held in the registers of each thread.
FragmentElement_ AccessType
The output type.
Definition: fragment_load_store.h:57
Definition: fragment_load_store.h:91
static CUTLASS_DEVICE void load(AccessType &value, Scalar_ const *pointer, int offset)
The load function.
Definition: fragment_load_store.h:60
Specifies whether iterator storage fragment consists of Scalar values or WMMA matrix.
Definition: tile_iterator.h:66