34 #if defined(CUTLASS_ARCH_WMMA_ENABLED)    50 class WmmaFragmentArray: 
public Array<T, N, true> {
    56     for(
int i=0; i<Array<T, N, true>::kElements; i++) {
    58       nvcuda::wmma::fill_fragment((*
this)[i], (
typename T::element_type)0);
    71 #endif // if defined(CUTLASS_ARCH_WMMA_ENABLED) Definition: aligned_buffer.h:35
Statically sized array of elements that accommodates all CUTLASS-supported numeric types and is safe ...
#define CUTLASS_HOST_DEVICE
Definition: cutlass.h:89
Templates exposing architecture support for warp matrix multiply-add (WMMA) operations. 
Basic include for CUTLASS.