Cutlass
CUDA Templates for Linear Algebra Subroutines and Solvers
|
Free functions for loading and storing to implementations of tile iteartor concepts. More...
#include <cutlass/fragment_load_store.h>
#include <cutlass/load_store.h>
#include <cutlass/predicate_vector.h>
#include <cutlass/shape.h>
Go to the source code of this file.
Namespaces | |
cutlass | |
Functions | |
template<typename InputIterator , typename Fragment > | |
CUTLASS_HOST_DEVICE void | cutlass::iterator_load (InputIterator &iterator, Fragment &fragment) |
Loads a fragment from an input iterator. More... | |
template<typename InputIterator , typename Fragment > | |
CUTLASS_DEVICE void | cutlass::shared_iterator_load (InputIterator &iterator, Fragment &fragment) |
Loads a fragment from a shared memory input iterator. More... | |
template<typename InputIterator , typename Fragment > | |
CUTLASS_DEVICE void | cutlass::shared_iterator_load (InputIterator &iterator, Fragment &fragment, int d) |
Loads a fragment from a shared memory input iterator. More... | |
template<typename InputIterator , typename Fragment , typename ConstPredicateAdapter > | |
CUTLASS_HOST_DEVICE void | cutlass::iterator_load_post_increment (InputIterator &iterator, Fragment &fragment, typename InputIterator::Index offset, ConstPredicateAdapter predicate_adapter) |
Loads a fragment from an input iterator, masked by a predicate iterator. More... | |
template<typename InputIterator , typename Fragment > | |
CUTLASS_HOST_DEVICE void | cutlass::iterator_load_post_increment (InputIterator &iterator, Fragment &fragment, typename InputIterator::Index offset=0) |
Loads a fragment from an input iterator. More... | |
template<typename InputIterator , typename Fragment , typename ConstPredicateAdapter > | |
CUTLASS_HOST_DEVICE void | cutlass::iterator_load_post_increment (InputIterator &iterator, Fragment &fragment, ConstPredicateAdapter pred_it) |
Loads a fragment from an input iterator. More... | |
template<typename InputIterator , typename Fragment , typename ConstPredicateAdapter > | |
CUTLASS_HOST_DEVICE void | cutlass::iterator_load (InputIterator const &_iterator, Fragment &fragment, typename InputIterator::Index offset, ConstPredicateAdapter predicate_adapter) |
template<typename InputIterator , typename Fragment > | |
CUTLASS_HOST_DEVICE void | cutlass::iterator_load (InputIterator const &iterator, Fragment &fragment, typename InputIterator::Index offset=0) |
Loads a fragment from an input iterator. More... | |
template<typename InputIterator , typename Fragment , typename ConstPredicateAdapter > | |
CUTLASS_HOST_DEVICE void | cutlass::iterator_load (InputIterator const &iterator, Fragment &fragment, ConstPredicateAdapter pred_it) |
Loads a fragment from an input iterator. More... | |
template<typename OutputIterator , typename Fragment > | |
CUTLASS_HOST_DEVICE void | cutlass::iterator_store (OutputIterator &iterator, Fragment &fragment) |
Stores a fragment to an output iterator. More... | |
template<typename OutputIterator , typename Fragment > | |
CUTLASS_DEVICE void | cutlass::shared_iterator_store (OutputIterator &iterator, Fragment const &fragment) |
Stores a fragment to a shared memory output iterator. More... | |
template<typename OutputIterator , typename Fragment , typename ConstPredicateAdapter > | |
CUTLASS_HOST_DEVICE void | cutlass::iterator_store_post_increment (OutputIterator &iterator, Fragment const &fragment, typename OutputIterator::Index offset, ConstPredicateAdapter predicate_adapter) |
Stores a fragment to an output iterator, masked by a predicate iterator. More... | |
template<typename OutputIterator , typename Fragment > | |
CUTLASS_HOST_DEVICE void | cutlass::iterator_store_post_increment (OutputIterator &iterator, Fragment const &fragment, typename OutputIterator::Index offset=0) |
Stores a fragment to an output iterator. More... | |
template<typename OutputIterator , typename Fragment , typename ConstPredicateAdapter > | |
CUTLASS_HOST_DEVICE void | cutlass::iterator_store_post_increment (OutputIterator &iterator, Fragment const &fragment, ConstPredicateAdapter pred_it) |
Stores a fragment to an output iterator. More... | |
template<typename OutputIterator , typename Fragment , typename ConstPredicateAdapter > | |
CUTLASS_HOST_DEVICE void | cutlass::iterator_store (OutputIterator const &_iterator, Fragment const &fragment, typename OutputIterator::Index offset, ConstPredicateAdapter predicate_adapter) |
Stores a fragment to an output iterator, masked by a predicate iterator. More... | |
template<typename OutputIterator , typename Fragment > | |
CUTLASS_HOST_DEVICE void | cutlass::iterator_store (OutputIterator const &iterator, Fragment const &fragment, typename OutputIterator::Index offset=0) |
Stores a fragment to an output iterator. More... | |
template<typename OutputIterator , typename Fragment , typename ConstPredicateAdapter > | |
CUTLASS_HOST_DEVICE void | cutlass::iterator_store (OutputIterator const &iterator, Fragment const &fragment, ConstPredicateAdapter pred_it) |
Stores a fragment to an output iterator. More... | |