Tidy up max_alignment TODO

codeplaysoftware · Dec 3, 2024 · 482bf0e · 482bf0e
1 parent 792e777
commit 482bf0e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/test/unit/gemm/device/gemm_testbed_3x.hpp b/test/unit/gemm/device/gemm_testbed_3x.hpp
@@ -1808,7 +1808,8 @@ bool TestXe(
   // For M & N we test a small and a big size
   // For K, we currently only support K = TileShapeK
   // We set L = 1 throughout
-  int max_alignment = 4; //std::max(Gemm::kAlignmentA, Gemm::kAlignmentB);
+  // TODO(codeplay): unhardcode max_alignment
+  int max_alignment = 4;
   std::vector<int> problem_size_m{max_alignment, 512 - 3 * max_alignment};
   std::vector<int> problem_size_n{max_alignment, 512 - 2 * max_alignment};