Skip to content

GitLab

"torchvision/models/vscode:/vscode.git/clone" did not exist on "7998cdfa56b87e69387db6152e0d02749f852499"
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
static constexpr index_t MRepeats = MRepeats_;
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()
{
......@@ -831,8 +835,8 @@ struct XdlopsGemm
static constexpr index_t MPerXdlops = GetXdlopsInfo().MPerXdlops;
static constexpr index_t NPerXdlops = GetXdlopsInfo().NPerXdlops;
static constexpr bool IsKReduction = GetXdlopsInfo().IsKReduction();
// static constexpr bool IsABroadcast = GetXdlopsInfo().IsABroadcast();
static constexpr bool IsKReduction = GetXdlopsInfo().IsKReduction();
static constexpr bool IsABroadcast = GetXdlopsInfo().IsABroadcast();
static constexpr index_t KPerXdlops = GetXdlopsInfo().GetKPerXdlops();
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