CUTLASS 3.8 Release (#2059)
* CUTLASS 3.8 Release * update * Update README.md * Revert "Update README.md" This reverts commit b353e36fe83e0815f99b44e46c0c95494c44726b. * update * update --------- Co-authored-by: Haicheng Wu <57973641+hwu36@users.noreply.github.com> Co-authored-by: Haicheng Wu <haichengw@nvidia.com>
This commit is contained in:
co-authored by
Haicheng Wu
Haicheng Wu
parent
9eb01fa0b0
commit
389e493055
@@ -679,6 +679,11 @@ struct GetName<cutlass::float_e4m3_t> {
|
||||
static constexpr char name[] = "float_e4m3_t";
|
||||
};
|
||||
|
||||
template <>
|
||||
struct GetName<cutlass::float_e5m2_t> {
|
||||
static constexpr char name[] = "float_e5m2_t";
|
||||
};
|
||||
|
||||
template <>
|
||||
struct GetName<cutlass::half_t> {
|
||||
static constexpr char name[] = "half_t";
|
||||
@@ -724,13 +729,20 @@ using VectorConvertTypes = ::testing::Types<
|
||||
ResultSourcePair<cutlass::bfloat16_t, uint8_t>,
|
||||
ResultSourcePair<cutlass::bfloat16_t, int8_t>,
|
||||
|
||||
ResultSourcePair<cutlass::float_e4m3_t, cutlass::int2b_t>,
|
||||
ResultSourcePair<cutlass::float_e5m2_t, cutlass::int2b_t>,
|
||||
ResultSourcePair<cutlass::half_t, cutlass::int2b_t>,
|
||||
ResultSourcePair<cutlass::bfloat16_t, cutlass::int2b_t>,
|
||||
ResultSourcePair<cutlass::float_e4m3_t, cutlass::uint2b_t>,
|
||||
ResultSourcePair<cutlass::float_e5m2_t, cutlass::uint2b_t>,
|
||||
ResultSourcePair<cutlass::half_t, cutlass::uint2b_t>,
|
||||
ResultSourcePair<cutlass::bfloat16_t, cutlass::uint2b_t>,
|
||||
|
||||
ResultSourcePair<cutlass::float_e4m3_t, cutlass::int4b_t>,
|
||||
ResultSourcePair<cutlass::float_e5m2_t, cutlass::int4b_t>,
|
||||
ResultSourcePair<cutlass::half_t, cutlass::int4b_t>,
|
||||
ResultSourcePair<cutlass::bfloat16_t, cutlass::int4b_t>,
|
||||
ResultSourcePair<cutlass::float_e4m3_t, cutlass::uint4b_t>,
|
||||
ResultSourcePair<cutlass::half_t, cutlass::uint4b_t>,
|
||||
ResultSourcePair<cutlass::bfloat16_t, cutlass::uint4b_t>,
|
||||
ResultSourcePair<float, cutlass::int4b_t>
|
||||
|
||||
Reference in New Issue
Block a user