Cutlass
CUDA Templates for Linear Algebra Subroutines and Solvers
|
#include <scalar_or_pointer.h>
Public Types | |
typedef Scalar_ | Scalar |
Underlying scalar type. More... | |
Public Member Functions | |
CUTLASS_HOST_DEVICE | ScalarOrPointer () |
Default ctor. More... | |
CUTLASS_HOST_DEVICE | ScalarOrPointer (Scalar const &val) |
Object behaves as a scalar. More... | |
CUTLASS_HOST_DEVICE | ScalarOrPointer (Scalar const *ptr_) |
Object behaves as a scalar. More... | |
CUTLASS_HOST_DEVICE bool | is_pointer () const |
Returns true if is pointer. More... | |
CUTLASS_HOST_DEVICE Scalar const * | get_ptr () const |
Gets the pointer value. More... | |
CUTLASS_HOST_DEVICE Scalar | get_scalar () const |
Gets the pointer value. More... | |
CUTLASS_HOST_DEVICE ScalarOrPointer & | operator= (Scalar const &scalar_) |
Assigns to a scalar and sets pointer to nullptr. More... | |
CUTLASS_HOST_DEVICE ScalarOrPointer & | operator= (Scalar const *ptr_) |
Assigns to a pointer value. More... | |
CUTLASS_HOST_DEVICE Scalar | get () const |
Access the element. More... | |
CUTLASS_HOST_DEVICE | operator Scalar () const |
Accesses the element. More... | |
Helper class defines an object which operates as either a scalar or a pointer. If the pointer is non-null, it is dereferenced when the object is accessed.
typedef Scalar_ cutlass::detail::ScalarOrPointer< Scalar_ >::Scalar |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |
|
inline |