Reorganize of KPack in GEMM (#3247)

* add the reorganize of KPack

* fix the compilation error

* fix the compilation error
This commit is contained in:
Thomas Ning
2025-11-24 12:38:59 -08:00
committed by GitHub
parent e95337c58c
commit de6a9590ab
4 changed files with 52 additions and 17 deletions

View File

@@ -120,7 +120,7 @@ int main(int argc, char* argv[])
#if CK_TILE_USE_WMMA
return !run_gemm_example<GemmConfigComputeV3_WMMA>(arg_parser);
#else
return !run_gemm_example<GemmConfigComputeV3>(arg_parser);
return !run_gemm_example<GemmConfigComputeV3_2>(arg_parser);
#endif
}
catch(const std::runtime_error& e)