43 typename Transformer_ = Copy<typename Iterator_::Fragment> >
61 typedef typename Iterator::Scalar
Scalar;
95 CUTLASS_DEVICE
void copy() {
CUTLASS_DEVICE void commit()
Commit the data.
Definition: gemm_shared_stream.h:103
Defines a structure containing strides, bounds, and a pointer to tensor data.
FetchedFragment fetched[2]
Fetched fragment.
Definition: gemm_shared_stream.h:124
CUTLASS_DEVICE void commit(int step)
Commit the data.
Definition: gemm_shared_stream.h:106
Iterator::Storage SharedStorage
The storage in shared memory needed by that stream.
Definition: gemm_shared_stream.h:76
CUTLASS_DEVICE void inc_stage()
Increment the stage.
Definition: gemm_shared_stream.h:119
TensorRef< Scalar, 4 > TensorRef
Reference type to a tensor.
Definition: gemm_shared_stream.h:64
CUTLASS_DEVICE SharedLoadStream()
Ctor.
Definition: gemm_shared_stream.h:79
TransformedFragment transformed[2]
Transformed fragment.
Definition: gemm_shared_stream.h:128
Defines iterators for efficiently loading and storing tiles to and from shared memory.
Definition: gemm_shared_stream.h:45
CUTLASS_DEVICE TransformedFragment const & fragment(int step=0) const
Returns the fragment for the given step.
Definition: gemm_shared_stream.h:114
Transformer transformer
The transformer.
Definition: gemm_shared_stream.h:126
TransformedFragment Fragment
Make sure the fragments match.
Definition: gemm_shared_stream.h:57
#define CUTLASS_HOST_DEVICE
Definition: cutlass.h:46
Iterator::Params iterator
The iterator params.
Definition: gemm_shared_stream.h:69
Iterator iterator
The iterator.
Definition: gemm_shared_stream.h:122
CUTLASS_HOST_DEVICE int initialize()
Setup the params.
Definition: gemm_shared_stream.h:72
Transformer::OutputFragment TransformedFragment
The fragment that is obtained after the transformation by the transformer.
Definition: gemm_shared_stream.h:54
CUTLASS_DEVICE SharedLoadStream(Params const ¶ms, TensorRef const &ref)
Ctor.
Definition: gemm_shared_stream.h:82
CUTLASS_DEVICE TransformedFragment & fragment(int step=0)
Returns the fragment for the given step.
Definition: gemm_shared_stream.h:111
CUTLASS_DEVICE void copy(int step)
Load the data from shared memory to the fetch fragment.
Definition: gemm_shared_stream.h:100
The params.
Definition: gemm_shared_stream.h:67
CUTLASS_DEVICE void initialize(Params const ¶ms, TensorRef const &ref)
Initialize the stream.
Definition: gemm_shared_stream.h:87
Iterator::Scalar Scalar
Scalar data type.
Definition: gemm_shared_stream.h:61
Iterator::Fragment FetchedFragment
The fragment that is copied from shared memory.
Definition: gemm_shared_stream.h:52
Transformer_ Transformer
The transformer.
Definition: gemm_shared_stream.h:49
Iterator_ Iterator
The load iterator.
Definition: gemm_shared_stream.h:47
CUTLASS_DEVICE void copy()
Load the data from shared memory to the fetch fragment.
Definition: gemm_shared_stream.h:95