From 99248f750755cf2ce37c25a61c567dc8011c3b47 Mon Sep 17 00:00:00 2001 From: Jacob Hinkle <1454944+jacobhinkle@users.noreply.github.com> Date: Wed, 29 Jan 2025 12:59:35 -0500 Subject: [PATCH] Apply suggestions from code review Co-authored-by: Ryan Spring --- csrc/scheduler/matmul_heuristic.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/csrc/scheduler/matmul_heuristic.h b/csrc/scheduler/matmul_heuristic.h index 00e91273e5e..c2889f2441f 100644 --- a/csrc/scheduler/matmul_heuristic.h +++ b/csrc/scheduler/matmul_heuristic.h @@ -208,7 +208,7 @@ class MatmulParams : public HeuristicParams { std::string toString() const { std::stringstream ss; - ss << x << " " << y << " " << z; + ss << "__cluster_dims__(" << x << ", " << "Y" << ", " << "Z" << ")"; return ss.str(); } @@ -241,7 +241,7 @@ class MatmulParams : public HeuristicParams { : "column-major") << "\n" << "Grid swizzle factor: " << grid_swizzle_factor << "\n" - << "Cluster dimensions: " << cluster_dims.toString() << "\n" + << cluster_dims.toString() << "\n" << "Use shared memory epilogue: " << use_smem_epilogue << "\n" << "Promote re-use of prologue shared memory: " << promote_prologue_smem_reuse << "\n"