tmp save

example/ck_tile/03_gemm/gemm_basic.cpp

@@ -103,11 +103,11 @@ float gemm(const gemm_traits& t, const ck_tile::GemmHostArgs& args, const ck_til
     using Row = ck_tile::tensor_layout::gemm::RowMajor;
     using Col = ck_tile::tensor_layout::gemm::ColumnMajor;
 
-    if(t.is_a_rowmajor && t.is_b_rowmajor && t.is_c_rowmajor)
-    {
-        return gemm_<Row, Row, Row>(args, s);
-    }
-    else if(t.is_a_rowmajor && !t.is_b_rowmajor && t.is_c_rowmajor)
+    // if(t.is_a_rowmajor && t.is_b_rowmajor && t.is_c_rowmajor)
+    // {
+    //     return gemm_<Row, Row, Row>(args, s);
+    // }
+    if(t.is_a_rowmajor && !t.is_b_rowmajor && t.is_c_rowmajor)
     {
         return gemm_<Row, Col, Row>(args, s);
     }

example/ck_tile/03_gemm/instances/gemm_universal_comp_instance_common.hpp

@@ -21,6 +21,9 @@ float gemm_(const ck_tile::GemmHostArgs& args, const ck_tile::stream_config& s)
                                                                      typename Traits_::CDataType,
                                                                      Traits_::kPadM,
                                                                      Traits_::kPadN>>;
+    constexpr bool TransposeC = false;
+    using GemmUniversalTraits = ck_tile::
+        TileGemmUniversalTraits<Traits_::kPadM, Traits_::kPadN, Traits_::kPadK, Traits_::ALayout, Traits_::BLayout, Traits_::CLayout, TransposeC>;
     using GemmTraits       = ck_tile::TileGemmTraits<Traits_::kPadM,
                                                Traits_::kPadN,
                                                Traits_::kPadK,
@@ -53,10 +56,10 @@ float gemm_(const ck_tile::GemmHostArgs& args, const ck_tile::stream_config& s)
                                                   typename Traits_::BDataType,
                                                   typename Traits_::AccDataType,
                                                   GemmShape,
-                                                  GemmTraits,
+                                                  GemmUniversalTraits,
                                                   ck_tile::GemmPipelineScheduler::Intrawave,
                                                   has_hot_loop_v,
-                                                  tail_number_v>>;
+                                                  tail_number_v>, ck_tile::UniversalGemmPipelineAgBgCrPolicy>;
         using Kernel = ck_tile::GemmKernel<TilePartitioner, GemmPipeline, GemmEpilogue>;
         auto kargs   = Kernel::MakeKernelArgs(args);