Commit f7f9954d authored by Harisankar Sadasivan's avatar Harisankar Sadasivan
Browse files

removed conflicts in struct members between streamk and unievrsal streamk

parent b469ec5f
...@@ -43,6 +43,18 @@ struct ProblemSizeStreamK final ...@@ -43,6 +43,18 @@ struct ProblemSizeStreamK final
ck::index_t StrideB = 4096; ck::index_t StrideB = 4096;
ck::index_t StrideC = 4096; ck::index_t StrideC = 4096;
ck::index_t NumSKBlocks = -1;
};
struct ProblemSizeStreamK_universal final
{
ck::index_t M = 3840;
ck::index_t N = 4096;
ck::index_t K = 4096;
ck::index_t StrideA = 4096;
ck::index_t StrideB = 4096;
ck::index_t StrideC = 4096;
ck::index_t Grid_size = -1; // defaults to max occupancy ck::index_t Grid_size = -1; // defaults to max occupancy
ck::index_t Streamk_sel = 1; // defaults to 1-tile SK ck::index_t Streamk_sel = 1; // defaults to 1-tile SK
}; };
......
...@@ -45,4 +45,4 @@ using ReferenceGemmInstance = ck::tensor_operation::host:: ...@@ -45,4 +45,4 @@ using ReferenceGemmInstance = ck::tensor_operation::host::
#include "run_gemm_example_streamk_v2.inc" #include "run_gemm_example_streamk_v2.inc"
int main(int argc, char* argv[]) { return !run_gemm_splitk_example(argc, argv); } int main(int argc, char* argv[]) { return !run_gemm_universal_streamk_example(argc, argv); }
...@@ -290,9 +290,9 @@ bool run_gemm(const ProblemType& problem_size, const ExecutionConfig& config) ...@@ -290,9 +290,9 @@ bool run_gemm(const ProblemType& problem_size, const ExecutionConfig& config)
return pass; return pass;
} }
bool run_gemm_splitk_example(int argc, char* argv[]) bool run_gemm_universal_streamk_example(int argc, char* argv[])
{ {
ProblemSizeStreamK problem_size; ProblemSizeStreamK_universal problem_size;
ExecutionConfig config; ExecutionConfig config;
return !parse_cmd_args(argc, argv, problem_size, config) || run_gemm(problem_size, config); return !parse_cmd_args(argc, argv, problem_size, config) || run_gemm(problem_size, config);
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment