[CI, CK examples] Disable time_kernel for CI tests and examples (#3464)

* Disable kernel timing in tests

* default time_kernel = false in old CK examples
This commit is contained in:
Johannes Graner
2026-01-07 16:30:57 +01:00
committed by GitHub
parent e8cc75aefb
commit 0a474aa62f
61 changed files with 61 additions and 61 deletions

View File

@@ -119,7 +119,7 @@ int main(int argc, char* argv[])
{
bool do_verification = true;
int init_method = 1;
bool time_kernel = true;
bool time_kernel = false;
// GEMM shape
ck::index_t M = 3840;

View File

@@ -119,7 +119,7 @@ int main(int argc, char* argv[])
{
bool do_verification = true;
int init_method = 1;
bool time_kernel = true;
bool time_kernel = false;
// GEMM shape
ck::index_t M = 3840;

View File

@@ -31,7 +31,7 @@ class SimpleAppArgs
bool do_verification = true;
int data_type = 1;
int init_method = 2;
bool time_kernel = true;
bool time_kernel = false;
public:
void show_usage(const char* cmd)

View File

@@ -31,7 +31,7 @@ class SimpleAppArgs
bool do_verification = true;
int data_type = 1;
int init_method = 2;
bool time_kernel = true;
bool time_kernel = false;
public:
void show_usage(const char* cmd)

View File

@@ -31,7 +31,7 @@ class SimpleAppArgs
bool do_verification = true;
int data_type = 1;
int init_method = 2;
bool time_kernel = true;
bool time_kernel = false;
public:
void show_usage(const char* cmd)

View File

@@ -53,7 +53,7 @@ int main(int argc, char* argv[])
{
do_verification = true;
init_method = 1;
time_kernel = true;
time_kernel = false;
}
else if(argc == 4)
{

View File

@@ -90,7 +90,7 @@ struct ExecutionConfig final
bool do_verification = true;
int init_method = 1;
int k_batch = 128;
bool time_kernel = true;
bool time_kernel = false;
};
bool run_grouped_gemm(const ProblemSize& problem_size, const ExecutionConfig& config)

View File

@@ -89,7 +89,7 @@ struct ExecutionConfig final
{
bool do_verification = true;
int init_method = 1;
bool time_kernel = true;
bool time_kernel = false;
};
bool run_grouped_gemm(const ProblemSize& problem_size, const ExecutionConfig& config)

View File

@@ -268,7 +268,7 @@ int main()
pass &= ck::utils::check_err(r1_m, r1_m_host, "Error: Incorrect results d1", 1e-2, 1e-2);
}
bool time_kernel = true;
bool time_kernel = false;
if(time_kernel)
{
float ave_time = invoker.Run(argument, StreamConfig{nullptr, time_kernel});

View File

@@ -302,7 +302,7 @@ int main(int argc, char* argv[])
{
bool do_verification = true;
int init_method = 1;
bool time_kernel = true;
bool time_kernel = false;
// GEMM shape
ck::index_t M = 1024;

View File

@@ -106,7 +106,7 @@ int main(int argc, char* argv[])
{
bool do_verification = true;
int init_method = 1;
bool time_kernel = true;
bool time_kernel = false;
// GEMM shape
ck::index_t M = 1024;

View File

@@ -106,7 +106,7 @@ int main(int argc, char* argv[])
{
bool do_verification = true;
int init_method = 1;
bool time_kernel = true;
bool time_kernel = false;
// GEMM shape
ck::index_t M = 1024;

View File

@@ -106,7 +106,7 @@ int main(int argc, char* argv[])
{
bool do_verification = true;
int init_method = 1;
bool time_kernel = true;
bool time_kernel = false;
// GEMM shape
ck::index_t M = 1024;

View File

@@ -108,7 +108,7 @@ int main(int argc, char* argv[])
{
bool do_verification = true;
int init_method = 1;
bool time_kernel = true;
bool time_kernel = false;
// GEMM shape
ck::index_t M = 1024;

View File

@@ -105,7 +105,7 @@ int main(int argc, char* argv[])
{
bool do_verification = true;
int init_method = 1;
bool time_kernel = true;
bool time_kernel = false;
// GEMM shape
ck::index_t M = 1024;

View File

@@ -112,7 +112,7 @@ int main(int argc, char* argv[])
{
bool do_verification = true;
int init_method = 1;
bool time_kernel = true;
bool time_kernel = false;
// GEMM shape
ck::index_t M = 1024;

View File

@@ -112,7 +112,7 @@ int main(int argc, char* argv[])
{
bool do_verification = true;
int init_method = 1;
bool time_kernel = true;
bool time_kernel = false;
// GEMM shape
ck::index_t M = 1024;

View File

@@ -112,7 +112,7 @@ int main(int argc, char* argv[])
{
bool do_verification = true;
int init_method = 1;
bool time_kernel = true;
bool time_kernel = false;
// GEMM shape
ck::index_t M = 1024;

View File

@@ -81,7 +81,7 @@ int main(int argc, char* argv[])
{
bool do_verification = true;
int init_method = 1;
bool time_kernel = true;
bool time_kernel = false;
// CGEMM shape
ck::index_t M = 1024;

View File

@@ -65,7 +65,7 @@ class SimpleAppArgs
bool do_verification = true;
int init_method = 2;
bool time_kernel = true;
bool time_kernel = false;
public:
void show_usage(const char* cmd)

View File

@@ -27,7 +27,7 @@ struct ExecutionConfig final
{
bool do_verification = true;
int init_method = 1;
bool time_kernel = true;
bool time_kernel = false;
};
template <typename DataType>

View File

@@ -248,7 +248,7 @@ int main(int argc, char* argv[])
{
bool do_verification = true;
int init_method = 1;
bool time_kernel = true;
bool time_kernel = false;
ck::index_t G0 = 1;
ck::index_t G1 = 2;

View File

@@ -92,7 +92,7 @@ struct ExecutionConfig final
{
bool do_verification = true;
int init_method = 1;
bool time_kernel = true;
bool time_kernel = false;
};
#define DefaultConvParam \

View File

@@ -92,7 +92,7 @@ struct ExecutionConfig final
{
bool do_verification = true;
int init_method = 1;
bool time_kernel = true;
bool time_kernel = false;
};
#define DefaultConvParam \

View File

@@ -40,7 +40,7 @@ class SimpleAppArgs
bool do_verification = true;
int init_method = 2;
bool time_kernel = true;
bool time_kernel = false;
public:
SimpleAppArgs()

View File

@@ -44,7 +44,7 @@ struct ExecutionConfig final
{
bool do_verification = true;
int init_method = 2;
bool time_kernel = true;
bool time_kernel = false;
};
template <ck::index_t... Is>

View File

@@ -56,7 +56,7 @@ template<> struct emb_kernel<ck::half_t, 8192> { using kernel_type = DeviceInsta
int main(int argc, char* argv[])
{
bool time_kernel = true;
bool time_kernel = false;
ck::index_t num_rows = 65536;
constexpr auto dims = ck::Sequence<256, 512, 768, 1024, 1536, 2048, 4096, 8192>{};

View File

@@ -195,7 +195,7 @@ int main(int argc, char* argv[])
{
bool do_verification = true;
int init_method = 1;
bool time_kernel = true;
bool time_kernel = false;
// GEMM shape
ck::index_t M = 1024;

View File

@@ -86,7 +86,7 @@ using DeviceGroupedConvNDFwdInstance =
int main(int argc, char* argv[])
{
bool do_verification = true;
bool time_kernel = true;
bool time_kernel = false;
if(argc == 1)
{

View File

@@ -84,7 +84,7 @@ using DeviceGroupedConvNDFwdInstance =
int main(int argc, char* argv[])
{
bool do_verification = true;
bool time_kernel = true;
bool time_kernel = false;
if(argc == 1)
{

View File

@@ -87,7 +87,7 @@ using DeviceGroupedConvNDFwdInstance =
int main(int argc, char* argv[])
{
bool do_verification = true;
bool time_kernel = true;
bool time_kernel = false;
if(argc == 1)
{

View File

@@ -84,7 +84,7 @@ using DeviceGroupedConvNDFwdInstance =
int main(int argc, char* argv[])
{
bool do_verification = true;
bool time_kernel = true;
bool time_kernel = false;
if(argc == 1)
{

View File

@@ -84,7 +84,7 @@ using DeviceGroupedConvNDFwdInstance =
int main(int argc, char* argv[])
{
bool do_verification = true;
bool time_kernel = true;
bool time_kernel = false;
if(argc == 1)
{

View File

@@ -90,7 +90,7 @@ using DeviceGroupedConvNDFwdInstance =
int main(int argc, char* argv[])
{
bool do_verification = true;
bool time_kernel = true;
bool time_kernel = false;
if(argc == 1)
{

View File

@@ -88,7 +88,7 @@ using DeviceGroupedConvNDFwdInstance =
int main(int argc, char* argv[])
{
bool do_verification = true;
bool time_kernel = true;
bool time_kernel = false;
if(argc == 1)
{

View File

@@ -88,7 +88,7 @@ using DeviceGroupedConvNDFwdInstance =
int main(int argc, char* argv[])
{
bool do_verification = true;
bool time_kernel = true;
bool time_kernel = false;
if(argc == 1)
{

View File

@@ -12,7 +12,7 @@ int run_groupnorm_fwd_example(int argc, char* argv[])
ck::index_t C = 128;
bool do_verification = true;
bool time_kernel = true;
bool time_kernel = false;
bool log_kernel = true;
if(argc == 1)

View File

@@ -53,7 +53,7 @@ using DeviceElementwisePermuteInstance = ck::tensor_operation::device::DeviceEle
int main(int argc, char* argv[])
{
bool do_verification = true;
bool time_kernel = true;
bool time_kernel = false;
std::vector<std::size_t> nchw = {16, 128, 32, 64};

View File

@@ -46,7 +46,7 @@ using DeviceElementwisePermuteInstance = ck::tensor_operation::device::DeviceEle
int main(int argc, char* argv[])
{
bool do_verification = true;
bool time_kernel = true;
bool time_kernel = false;
if(argc == 1)
{

View File

@@ -50,7 +50,7 @@ using DeviceElementwisePermuteInstance = ck::tensor_operation::device::DeviceEle
int main(int argc, char* argv[])
{
bool do_verification = true;
bool time_kernel = true;
bool time_kernel = false;
if(argc == 1)
{

View File

@@ -50,7 +50,7 @@ using DeviceElementwisePermuteInstance = ck::tensor_operation::device::DeviceEle
int main(int argc, char* argv[])
{
bool do_verification = true;
bool time_kernel = true;
bool time_kernel = false;
if(argc == 1)
{

View File

@@ -49,7 +49,7 @@ using DeviceElementwisePermuteInstance = ck::tensor_operation::device::DeviceEle
int main(int argc, char* argv[])
{
bool do_verification = true;
bool time_kernel = true;
bool time_kernel = false;
if(argc == 1)
{

View File

@@ -50,7 +50,7 @@ using DeviceElementwisePermuteInstance = ck::tensor_operation::device::DeviceEle
int main(int argc, char* argv[])
{
bool do_verification = true;
bool time_kernel = true;
bool time_kernel = false;
if(argc == 1)
{

View File

@@ -121,7 +121,7 @@ void reference_scale_permute_amax(Tensor<InputDataType>& input,
int main(int argc, char* argv[])
{
bool do_verification = true;
bool time_kernel = true;
bool time_kernel = false;
const float scale = 2.f;

View File

@@ -58,7 +58,7 @@ using DeviceElementwisePermuteInstance = ck::tensor_operation::device::DeviceEle
int main(int argc, char* argv[])
{
bool do_verification = true;
bool time_kernel = true;
bool time_kernel = false;
if(argc == 1)
{

View File

@@ -84,7 +84,7 @@ void host_elementwise2D(HostTensorC& C,
int main(int argc, char* argv[])
{
bool do_verification = true;
bool time_kernel = true;
bool time_kernel = false;
ck::index_t M = 48 * 256;
ck::index_t N = 1024;

View File

@@ -205,7 +205,7 @@ int main(int argc, char* argv[])
{
bool do_verification = true;
int init_method = 1;
bool time_kernel = true;
bool time_kernel = false;
// GEMM shape
ck::index_t N = 4096;

View File

@@ -193,7 +193,7 @@ int main(int argc, char* argv[])
{
bool do_verification = true;
int init_method = 1;
bool time_kernel = true;
bool time_kernel = false;
#if 1
// GEMM shape
ck::index_t N = 4096;

View File

@@ -194,7 +194,7 @@ int main(int argc, char* argv[])
{
bool do_verification = true;
int init_method = 1;
bool time_kernel = true;
bool time_kernel = false;
// per expert:
// GEMM shape

View File

@@ -185,7 +185,7 @@ int main(int argc, char* argv[])
{
bool do_verification = true;
int init_method = 1;
bool time_kernel = true;
bool time_kernel = false;
// per expert:
// GEMM shape

View File

@@ -188,7 +188,7 @@ int main(int argc, char* argv[])
{
bool do_verification = true;
int init_method = 1;
bool time_kernel = true;
bool time_kernel = false;
// tokens = 1
// topk = 1

View File

@@ -164,7 +164,7 @@ int main(int argc, char* argv[])
{
bool do_verification = true;
int init_method = 1;
bool time_kernel = true;
bool time_kernel = false;
// per expert:
// GEMM shape

View File

@@ -178,7 +178,7 @@ int main(int argc, char* argv[])
{
bool do_verification = true;
int init_method = 1;
bool time_kernel = true;
bool time_kernel = false;
// per expert:
// GEMM shape

View File

@@ -178,7 +178,7 @@ int main(int argc, char* argv[])
{
bool do_verification = true;
int init_method = 1;
bool time_kernel = true;
bool time_kernel = false;
// per expert:
// GEMM shape

View File

@@ -208,7 +208,7 @@ int main(int argc, char* argv[])
{
bool do_verification = true;
int init_method = 1;
bool time_kernel = true;
bool time_kernel = false;
// per expert:
// GEMM shape

View File

@@ -171,7 +171,7 @@ int main(int argc, char* argv[])
{
bool do_verification = true;
int init_method = 1;
bool time_kernel = true;
bool time_kernel = false;
// per expert:
// GEMM shape

View File

@@ -171,7 +171,7 @@ int main(int argc, char* argv[])
{
bool do_verification = true;
int init_method = 1;
bool time_kernel = true;
bool time_kernel = false;
// per expert:
// GEMM shape

View File

@@ -204,7 +204,7 @@ int main(int argc, char* argv[])
{
bool do_verification = true;
int init_method = 1;
bool time_kernel = true;
bool time_kernel = false;
// per expert:
// GEMM shape

View File

@@ -61,7 +61,7 @@ class TestBatchedGemmMultiD : public ::testing::Test
true, // do_verification
1, // init_method
false, // do_log
1, // time_kernel,
false, // time_kernel,
M,
N,
K,

View File

@@ -104,7 +104,7 @@ int main(int argc, char* argv[])
};
bool do_verification = true;
bool time_kernel = true;
bool time_kernel = false;
int problem_index = -1;
if(argc == 1)

View File

@@ -306,7 +306,7 @@ void PerformGemm(const ck::index_t M,
const auto kernel =
DeviceGemm<DataType, GemmTraits, scalar_per_vector, BlockShape, ThreadLayout, DoPadding>;
const float avg_time = launch_and_time_kernel(StreamConfig{nullptr, true},
const float avg_time = launch_and_time_kernel(StreamConfig{nullptr, false},
kernel,
dim3(grid_size_x, grid_size_y, 1),
dim3(ck::wrapper::size(thread_layout)),