41 template <
typename Scalar_>
CUTLASS_HOST_DEVICE Scalar get_scalar() const
Gets the pointer value.
Definition: scalar_or_pointer.h:90
CUTLASS_HOST_DEVICE ScalarOrPointer(Scalar const &val)
Object behaves as a scalar.
Definition: scalar_or_pointer.h:70
Scalar_ Scalar
Underlying scalar type.
Definition: scalar_or_pointer.h:45
CUTLASS_HOST_DEVICE ScalarOrPointer & operator=(Scalar const *ptr_)
Assigns to a pointer value.
Definition: scalar_or_pointer.h:104
#define CUTLASS_HOST_DEVICE
Definition: cutlass.h:46
CUTLASS_HOST_DEVICE ScalarOrPointer(Scalar const *ptr_)
Object behaves as a scalar.
Definition: scalar_or_pointer.h:74
CUTLASS_HOST_DEVICE bool is_pointer() const
Returns true if is pointer.
Definition: scalar_or_pointer.h:78
CUTLASS_HOST_DEVICE ScalarOrPointer()
Default ctor.
Definition: scalar_or_pointer.h:66
CUTLASS_HOST_DEVICE Scalar const * get_ptr() const
Gets the pointer value.
Definition: scalar_or_pointer.h:84
Basic include for CUTLASS macros.
CUTLASS_HOST_DEVICE ScalarOrPointer & operator=(Scalar const &scalar_)
Assigns to a scalar and sets pointer to nullptr.
Definition: scalar_or_pointer.h:96
Definition: scalar_or_pointer.h:42