fix wrong A/BLayout in MMA_Traits for binary mma and append other MMA_Traits support (#1856)

* fix wrong A/BLayout in  MMA_Traits<SM80_16x8x256_S32U1U1S32_TN_XORPOPC> and append support for  m8n8k128, m16n8k128  mma.and.popc in MMA_Traits instantiation

* add "print" template for  subbyte_reference<T>
This commit is contained in:
Caleb_Du
2024-10-25 02:38:35 +08:00
committed by GitHub
parent be692b48b0
commit a424ca6cf9
3 changed files with 155 additions and 4 deletions

View File

@@ -346,6 +346,11 @@ print(subbyte_iterator<T> const& x) {
printf("subptr[%db](%p.%u)", int(sizeof_bits_v<T>), x.ptr_, x.idx_);
}
template <class T>
CUTE_HOST_DEVICE void
print(subbyte_reference<T> const& x) {
print(x.get());
}
//
// array_subbyte
// Statically sized array for non-byte-aligned data types