ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-17 11:30:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge commit '515e28309153ae8ab6fa3cbed81b44e2c01c43cd' into develop

Browse Source
This commit is contained in:
assistant-librarian[bot]
2025-10-28 18:16:26 +00:00
parent 3b6a825599
commit d79c92527f
32 changed files with 7745 additions and 151 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
example/45_elementwise_normalization/elementwise_layernorm_blockwise.cpp
View File

@@ -82,37 +82,29 @@ int main(int argc, char* argv[])
bool do_verification = true;
bool time_kernel = true;
ck::index_t M = 48 * 256;
ck::index_t N = 1024;
if(argc == 1)
{
// use default
}
else if(argc == 3)
else if(argc == 3 || argc == 5)
{
do_verification = std::stoi(argv[1]);
time_kernel = std::stoi(argv[2]);
if(argc == 5)
{
M = std::stoi(argv[3]);
N = std::stoi(argv[4]);
}
}
else
{
printf("arg1: verification (0=no, 1=yes)\n");
printf("arg2: time kernel (0=no, 1=yes)\n");
exit(0);
}
ck::index_t M = 48 * 256;
ck::index_t N = 1024;
if(argc == 1)
{
// use default case
}
else if(argc == 3)
{
M = std::stoi(argv[1]);
N = std::stoi(argv[2]);
}
else
{
std::cerr << "arg1 to 2: M, N" << std::endl;
return 1;
printf("arg3-4: M, N\n");
exit(1);
}
ck::index_t Stride = N;