From 7549e2b2e6c35699d69e9c614bb2df5a91a59ad2 Mon Sep 17 00:00:00 2001 From: Yanxing-Shi Date: Mon, 26 May 2025 06:45:36 +0000 Subject: [PATCH] fix readme --- tile_engine/ops/gemm/README.md | 2 ++ tile_engine/ops/gemm/gemm_host_api.hpp | 8 ++++---- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/tile_engine/ops/gemm/README.md b/tile_engine/ops/gemm/README.md index 87267f8bce..4b04453392 100644 --- a/tile_engine/ops/gemm/README.md +++ b/tile_engine/ops/gemm/README.md @@ -36,6 +36,8 @@ rm -rf tile_engine/ && ninja benchmark_gemm # rebuild -stride_b The stride value for tensor B. Default is 0. -stride_c The stride value for tensor C Default is 0. -split_k The split value for k dimension. Default is 1. + -enable_profile_cache Whether flush profile cache or not when benchmark kernel. Possible values are or false. Default is true. + -flush_profile_cache Whether flush profile cache or not when benchmark kernel. Possible values are true or false. Default is false. -v The type of validation. Set to 0 for no validation, 1 for validation on CPU, or 2 for validation on GPU. Default is 2, validation on GPU. -log Wether output kernel instance information or not. Possible values are true or false. Default is false. -warmup The number of iterations before benchmark the kernel. Default is 50. diff --git a/tile_engine/ops/gemm/gemm_host_api.hpp b/tile_engine/ops/gemm/gemm_host_api.hpp index 280e228190..d26bffe2bb 100755 --- a/tile_engine/ops/gemm/gemm_host_api.hpp +++ b/tile_engine/ops/gemm/gemm_host_api.hpp @@ -75,12 +75,12 @@ inline auto create_args(int argc, char* argv[]) .insert("split_k", "1", "The split value for k dimension. Default is 1.") .insert("enable_profile_cache", "true", - "whether use profile cache or not when benchmark kernel, Possible values are true " - "or false. Default is true") + "Whether use profile cache or not when benchmark kernel, Possible values are true " + "or false. Default is true.") .insert("flush_profile_cache", "false", - "whether flush profile cache or not when benchmark kernel. Possible values are " - "true or false. Default is false") + "Whether flush profile cache or not when benchmark kernel. Possible values are " + "true or false. Default is false.") .insert("verify", "2", "The type of validation. Set to 0 for no validation, 1 for validation on CPU, or 2 "