"git@developer.sourcefind.cn:orangecat/ollama.git" did not exist on "b9bb5ca288523338729fcd4687e654096f635eb6"
Unverified Commit ae5e5181 authored by zjing14's avatar zjing14 Committed by GitHub
Browse files

recover default niter (#1064)

parent 7965d66a
...@@ -11,6 +11,6 @@ struct StreamConfig ...@@ -11,6 +11,6 @@ struct StreamConfig
hipStream_t stream_id_ = nullptr; hipStream_t stream_id_ = nullptr;
bool time_kernel_ = false; bool time_kernel_ = false;
int log_level_ = 0; int log_level_ = 0;
int cold_niters_ = 50; int cold_niters_ = 1;
int nrepeat_ = 200; int nrepeat_ = 10;
}; };
...@@ -166,7 +166,7 @@ int profile_gemm_impl(int do_verification, ...@@ -166,7 +166,7 @@ int profile_gemm_impl(int do_verification,
std::string op_name = op_ptr->GetTypeString(); std::string op_name = op_ptr->GetTypeString();
float avg_time = float avg_time =
invoker_ptr->Run(argument_ptr.get(), StreamConfig{nullptr, time_kernel}); invoker_ptr->Run(argument_ptr.get(), StreamConfig{nullptr, time_kernel, 0, 10, 50});
std::size_t flop = std::size_t(2) * M * N * K; std::size_t flop = std::size_t(2) * M * N * K;
......
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