Skip to content

GitLab

"...git@developer.sourcefind.cn:OpenDAS/opencompass.git" did not exist on "b48d084020669857cf99cc475090836e7807c6bf"
Commit 028bb4b6 authored by Paul's avatar Paul
Browse files

Use the right template

parent e5080ac5
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
host/test/gemm_multiple_d.cpp
View file @ 028bb4b6
......@@ -26,9 +26,9 @@ const std::string gemm_compile_check = R"__ck__(
extern "C" __global__ void f(const ck::half_t* a, const ck::half_t* b, ck::half_t* c) {
using G = ${template};
constexpr auto desc = ${template}::make_descriptor(ck::make_naive_tensor_descriptor_packed(ck::make_tuple(${m}, ${k})),
ck::make_naive_tensor_descriptor_packed(ck::make_tuple(${n, ${k})),
ck::make_naive_tensor_descriptor_packed(ck::make_tuple(${n}, ${k})),
ck::make_tuple(),
ck::make_naive_tensor_descriptor_packed(ck::make_tuple(${m, ${n})));
ck::make_naive_tensor_descriptor_packed(ck::make_tuple(${m}, ${n})));
static_assert(desc.IsValid(), "Invalid ck gemm.");
......@@ -49,7 +49,7 @@ TEST_CASE(test_problem_kernel)
prob.K = 256;
for(auto solution : prob.GetSolutions("gfx90a"))
{
auto src = ck::host::InterpolateString(compile_check,
auto src = ck::host::InterpolateString(gemm_compile_check,
{{"include", prob.GetIncludeHeader()},
{"template", solution.ToTemplateString()},
{"m", std::to_string(prob.M)},
......
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