Skip to content

GitLab

Commit 21755b5d authored by Jing Zhang's avatar Jing Zhang
Browse files

abroadcast only

parent de9f5bed
Hide whitespace changes
Inline Side-by-side
Showing with 7 additions and 3 deletions
composable_kernel/include/tensor_operation/xdlops_gemm.hpp
View file @ 21755b5d
...@@ -538,7 +538,11 @@ struct xdlops_info ...@@ -538,7 +538,11 @@ struct xdlops_info
static constexpr index_t MRepeats = MRepeats_; static constexpr index_t MRepeats = MRepeats_;
static constexpr index_t NRepeats = NRepeats_; static constexpr index_t NRepeats = NRepeats_;
// static constexpr bool IsABroadcast() { return NPerXdlops >= MPerXdlops; } static constexpr bool IsABroadcast()
{
static_assert(NPerXdlops >= MPerXdlops, "only support ABroadcast");
return true;
}
static constexpr bool IsKReduction() static constexpr bool IsKReduction()
{ {
...@@ -831,8 +835,8 @@ struct XdlopsGemm ...@@ -831,8 +835,8 @@ struct XdlopsGemm
static constexpr index_t MPerXdlops = GetXdlopsInfo().MPerXdlops; static constexpr index_t MPerXdlops = GetXdlopsInfo().MPerXdlops;
static constexpr index_t NPerXdlops = GetXdlopsInfo().NPerXdlops; static constexpr index_t NPerXdlops = GetXdlopsInfo().NPerXdlops;
static constexpr bool IsKReduction = GetXdlopsInfo().IsKReduction(); static constexpr bool IsKReduction = GetXdlopsInfo().IsKReduction();
// static constexpr bool IsABroadcast = GetXdlopsInfo().IsABroadcast(); static constexpr bool IsABroadcast = GetXdlopsInfo().IsABroadcast();
static constexpr index_t KPerXdlops = GetXdlopsInfo().GetKPerXdlops(); static constexpr index_t KPerXdlops = GetXdlopsInfo().GetKPerXdlops();
static constexpr auto GetBlkId(const index_t lane_id) static constexpr auto GetBlkId(const index_t lane_id)
......
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