Apply suggestions from code review

Co-authored-by: Ryan Spring <rspring@nvidia.com>
NVIDIA · Jan 29, 2025 · 99248f7 · 99248f7
1 parent fce996d
commit 99248f7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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"