Commit f0e7bc99 authored by Jing Zhang's avatar Jing Zhang
Browse files

enable fp16 mfma

parent cf378e46
...@@ -506,49 +506,49 @@ struct MfmaSelector ...@@ -506,49 +506,49 @@ struct MfmaSelector
return MfmaInstr::mfma_f32_16x16x4xf32; return MfmaInstr::mfma_f32_16x16x4xf32;
} }
#if 0
template <> template <>
static constexpr auto GetMfma<half_t, 64, 64>() static constexpr auto GetMfma<half_t, 64, 64>()
{ {
return xdlops_info<MfmaInstr::mfma_f32_32x32x4f16, 64, 64>{}; return MfmaInstr::mfma_f32_32x32x4f16;
} }
template <> template <>
static constexpr auto GetMfma<half_t, 32, 64>() static constexpr auto GetMfma<half_t, 32, 64>()
{ {
return xdlops_info<MfmaInstr::mfma_f32_32x32x4f16, 32, 64>{}; return MfmaInstr::mfma_f32_32x32x4f16;
} }
template <> template <>
static constexpr auto GetMfma<half_t, 32, 32>() static constexpr auto GetMfma<half_t, 32, 32>()
{ {
return xdlops_info<MfmaInstr::mfma_f32_32x32x8f16, 32, 32>{}; return MfmaInstr::mfma_f32_32x32x8f16;
} }
template <> template <>
static constexpr auto GetMfma<half_t, 16, 16>() static constexpr auto GetMfma<half_t, 16, 16>()
{ {
return xdlops_info<MfmaInstr::mfma_f32_16x16x16f16, 16, 16>{}; return MfmaInstr::mfma_f32_16x16x16f16;
} }
template <> template <>
static constexpr auto GetMfma<half_t, 16, 64>() static constexpr auto GetMfma<half_t, 16, 64>()
{ {
return xdlops_info<MfmaInstr::mfma_f32_16x16x4f16, 16, 64>{}; return MfmaInstr::mfma_f32_16x16x4f16;
} }
template <> template <>
static constexpr auto GetMfma<half_t, 8, 64>() static constexpr auto GetMfma<half_t, 8, 64>()
{ {
return xdlops_info<MfmaInstr::mfma_f32_4x4x4f16, 8, 64>{}; return MfmaInstr::mfma_f32_4x4x4f16;
} }
template <> template <>
static constexpr auto GetMfma<half_t, 4, 64>() static constexpr auto GetMfma<half_t, 4, 64>()
{ {
return xdlops_info<MfmaInstr::mfma_f32_4x4x4f16, 4, 64>{}; return MfmaInstr::mfma_f32_4x4x4f16;
} }
#if 0
template <> template <>
static constexpr auto GetMfma<ushort, 128, 64>() static constexpr auto GetMfma<ushort, 128, 64>()
{ {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment