ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-13 02:27:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

topk accuracy debug

Browse Source
This commit is contained in:
huizzhan
2025-08-07 08:45:43 +00:00
parent c1b0b3f835
commit e73c5da118
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
example/ck_tile/39_gemm_softmax_grouped_topk/gemm_softmax_grouped_topk.cpp
View File

@@ -254,6 +254,11 @@ int main(int argc, char* argv[])
std::string("] Value Error:"),
rtol,
atol);
// for(std::size_t i = 0; i < s_value_ref.size(); ++i) {
// const double o = *std::next(std::begin(s_value_host), i);
// const double r = *std::next(std::begin(s_value_ref), i);
// std::cout << " out[" << i << "] != ref[" << i << "]: " << o << " != " << r << std::endl;
// }
auto s_index_host = index_host_dev.slice(s_begin, s_end);
auto s_index_ref = index_ref.slice(s_begin, s_end);
rtn &= ck_tile::check_err(s_index_host,
@@ -262,6 +267,11 @@ int main(int argc, char* argv[])
std::string("] Index Error:"),
rtol,
atol);
// for(std::size_t i = 0; i < s_index_ref.size(); ++i) {
// const double o = *std::next(std::begin(s_index_host), i);
// const double r = *std::next(std::begin(s_index_ref), i);
// std::cout << " out[" << i << "] != ref[" << i << "]: " << o << " != " << r << std::endl;
// }
}
std::cout << "valid:" << (rtn ? "y" : "n") << std::endl;
}