reduce inital number for half_t splitk (#685)

8b9cbba8 · zjing14 · GitHub · 3eecbfb6 · 8b9cbba8
Unverified Commit 8b9cbba8 authored Apr 24, 2023 by zjing14 Committed by GitHub Apr 24, 2023
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

profiler/include/profiler/profile_gemm_splitk_impl.hpp profiler/include/profiler/profile_gemm_splitk_impl.hpp +2 -2

No files found.
--- a/profiler/include/profiler/profile_gemm_splitk_impl.hpp
+++ b/profiler/include/profiler/profile_gemm_splitk_impl.hpp
@@ -72,8 +72,8 @@ bool profile_gemm_splitk_impl(int do_verification,
    {
    case 0: break;
    case 1:
-        a_m_k.GenerateTensorValue(GeneratorTensor_2<ADataType>{-5, 5});
-        b_k_n.GenerateTensorValue(GeneratorTensor_2<BDataType>{-5, 5});
+        a_m_k.GenerateTensorValue(GeneratorTensor_2<ADataType>{0, 1});
+        b_k_n.GenerateTensorValue(GeneratorTensor_2<BDataType>{-1, 1});
        break;
    default:
        a_m_k.GenerateTensorValue(GeneratorTensor_3<ADataType>{0.0, 1.0});