CUTLASS
CUDA Templates for Linear Algebra Subroutines and Solvers
|
#include <subbyte_reference.h>
Public Types | |
using | Element = Element_ |
using | Storage = Storage_ |
using | StoragePointer = Storage * |
Public Member Functions | |
CUTLASS_HOST_DEVICE | SubbyteReference () |
CUTLASS_HOST_DEVICE | SubbyteReference (Element *ptr, int64_t offset) |
Constructor. More... | |
CUTLASS_HOST_DEVICE | SubbyteReference (Element *ptr=nullptr) |
Constructor. More... | |
CUTLASS_HOST_DEVICE StoragePointer | storage_pointer () const |
Gets storage pointer. More... | |
CUTLASS_HOST_DEVICE int | element_offset () const |
Gets element offset within storage vector. More... | |
CUTLASS_HOST_DEVICE Element | get () const |
Unpacks an element from memory. More... | |
CUTLASS_HOST_DEVICE SubbyteReference & | set (Element const &x) |
Stores an element to memory. More... | |
CUTLASS_HOST_DEVICE | operator Element () const |
Unpacks an element from memory. More... | |
CUTLASS_HOST_DEVICE SubbyteReference & | operator= (Element const &x) |
Stores an element to memory. More... | |
CUTLASS_HOST_DEVICE SubbyteReference & | operator= (SubbyteReference const &x) |
Stores an element to memory. More... | |
CUTLASS_HOST_DEVICE SubbyteReference & | operator= (ConstSubbyteReference< Element, Storage > const &x) |
Stores an element to memory. More... | |
CUTLASS_HOST_DEVICE SubbyteReference & | operator+= (int offset) |
Adds an offset in units of elements to the reference. More... | |
CUTLASS_HOST_DEVICE SubbyteReference & | operator+= (long long offset) |
Adds an offset in units of elements to the reference. More... | |
CUTLASS_HOST_DEVICE SubbyteReference & | operator-= (int offset) |
Adds an offset in units of elements to the reference. More... | |
CUTLASS_HOST_DEVICE SubbyteReference & | operator-= (long long offset) |
Adds an offset in units of elements to the reference. More... | |
CUTLASS_HOST_DEVICE SubbyteReference | operator+ (int offset) const |
Returns a reference to an element with a given offset from the current reference. More... | |
CUTLASS_HOST_DEVICE SubbyteReference | operator+ (long long offset) const |
Returns a reference to an element with a given offset from the current reference. More... | |
CUTLASS_HOST_DEVICE SubbyteReference | operator- (int offset) const |
Returns a reference to an element with a given offset from the current reference. More... | |
CUTLASS_HOST_DEVICE SubbyteReference | operator-= (long long offset) const |
Returns a reference to an element with a given offset from the current reference. More... | |
CUTLASS_HOST_DEVICE ptrdiff_t | operator- (SubbyteReference ref) const |
Computes the difference in elements between references. More... | |
CUTLASS_HOST_DEVICE | operator int () const |
Explicit cast to int. More... | |
CUTLASS_HOST_DEVICE | operator int64_t () const |
Explicit cast to signed 64-bit integer. More... | |
CUTLASS_HOST_DEVICE | operator uint64_t () const |
Explicit cast to unsigned 64-bit integer. More... | |
CUTLASS_HOST_DEVICE | operator float () const |
Explicit cast to float. More... | |
CUTLASS_HOST_DEVICE | operator double () const |
Explicit cast to double. More... | |
using cutlass::SubbyteReference< Element_, Storage_ >::Element = Element_ |
using cutlass::SubbyteReference< Element_, Storage_ >::Storage = Storage_ |
using cutlass::SubbyteReference< Element_, Storage_ >::StoragePointer = Storage * |
|
inline |
|
inline |
offset | pointer to memory logical offset in units of Element |
|
inline |
|
inline |
|
inline |
|
inlineexplicit |
|
inline |
|
inlineexplicit |
|
inlineexplicit |
|
inlineexplicit |
|
inlineexplicit |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |