Searched refs:ElementInputE (Results 1 – 4 of 4) sorted by relevance
26 static constexpr int kStrideBlock32x32 = (32 * 32) / (sizeof(ElementInputE) * 8);28 ElementInputE* _meta;29 ElementInputE* _meta_trans;37 return (rows * cols) / (8 * sizeof(ElementInputE));66 _meta = (ElementInputE*)metaN.data_ptr();67 _meta_trans = (ElementInputE*)metaT.data_ptr();94 ElementInputE* get_metaN(103 ElementInputE* get_metaT(118 ElementInputE* _meta;119 ElementInputE* _meta_trans;[all …]
18 using ElementInputE = uint16_t; variable104 ElementInputE* metadata_quad,416 ElementInputE* packed_meta_reordered = metadata_gmem.get_metaN( in sparse_semi_structured_tile_kernel()424 ElementInputE* packed_trans_meta_reordered = metadata_gmem.get_metaT( in sparse_semi_structured_tile_kernel()
169 using ElementInputE = typename Gemm::ElementE;190 switch (sizeof(ElementInputE)) {243 cutlass::TensorRef<ElementInputE, ReorderedLayoutInputE>(244 (ElementInputE*)meta.data_ptr(),
161 using ElementInputE = typename Gemm::ElementE;182 switch (sizeof(ElementInputE)) {208 cutlass::TensorRef<ElementInputE, ReorderedLayoutInputE>(209 (ElementInputE*)tensor_e.data_ptr(),