Universal gemm flush cache (#1251)

* add flush cache to device op

* add flush cache parameter to ckProfiler

* change calculate size a and b method

* chang evaluation time method foro AVERAGE to MEDIAN

* format code

* adjust some code

* fix core dumped

* remove loop call flush icache in kernel

* remove loop(outer) call flush icache

---------

Co-authored-by: letaoqin <letaoqin@amd.com>

[ROCm/composable_kernel commit: f448d179b7]
This commit is contained in:
ltqin
2024-04-26 04:07:14 +08:00
committed by GitHub
parent 2f333b5225
commit fef66ea961
6 changed files with 331 additions and 15 deletions

View File

@@ -33,7 +33,7 @@ enum struct GemmDataType
int profile_gemm_universal(int argc, char* argv[])
{
if(argc != 15 && argc != 17)
if(argc != 15 && argc != 18)
{
printf("arg1: tensor operation (" OP_NAME ": " OP_DESC ")\n");
printf("arg2: data type (0: fp32; 1: fp16; 2: bf16; 3: int8; 4: f8@f16; 5: f16@f8; 6: f16, "
@@ -51,6 +51,7 @@ int profile_gemm_universal(int argc, char* argv[])
printf("optional:\n");
printf("arg15: number of warm-up cycles (default 1)\n");
printf("arg16: number of iterations (default 10)\n");
printf("arg17: memory for rotating buffer (default 0, size in MB)\n");
exit(1);
}
@@ -70,12 +71,14 @@ int profile_gemm_universal(int argc, char* argv[])
const int StrideC = std::stoi(argv[13]);
const int KBatch = std::stoi(argv[14]);
int n_warmup = 1;
int n_iter = 10;
if(argc == 17)
int n_warmup = 1;
int n_iter = 10;
uint64_t rotating = 0;
if(argc == 18)
{
n_warmup = std::stoi(argv[15]);
n_iter = std::stoi(argv[16]);
rotating = std::stoull(argv[17]) * 1024 * 1024;
}
using F32 = float;
@@ -124,7 +127,8 @@ int profile_gemm_universal(int argc, char* argv[])
(StrideC < 0) ? DefaultStrideC : StrideC,
KBatch,
n_warmup,
n_iter);
n_iter,
rotating);
return pass ? 0 : 1;
};