CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers
|
Defines a structure containing strides, bounds, and a pointer to tensor data. More...
#include "cutlass/cutlass.h"
#include "cutlass/coord.h"
#include "cutlass/platform/platform.h"
#include "cutlass/subbyte_reference.h"
Go to the source code of this file.
Classes | |
class | cutlass::IdentityTensorLayout< Rank > |
class | cutlass::TensorRef< Element_, Layout_ > |
Namespaces | |
cutlass | |
Functions | |
template<typename Element , typename Layout > | |
CUTLASS_HOST_DEVICE TensorRef< Element, Layout > | cutlass::make_TensorRef (Element *ptr, Layout const &layout) |
Constructs a TensorRef, deducing types from arguments. More... | |
template<typename Element , typename Layout > | |
bool | cutlass::TensorRef_aligned (TensorRef< Element, Layout > const &ref, int alignment) |