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

Merge commit 'ffe9775e7071d524caa70cf13017ebe80fe6272b' into develop

Browse Source
This commit is contained in:
assistant-librarian[bot]
2025-09-11 20:11:33 +00:00
parent bc3e5d8191
commit ed41349e19
52 changed files with 633 additions and 122 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
example/45_elementwise_normalization/elementwise_layernorm_blockwise.cpp
View File

@@ -77,9 +77,26 @@ void host_elementwise2D(HostTensorC& C,
}
}
int main()
int main(int argc, char* argv[])
{
bool time_kernel = true;
bool do_verification = true;
bool time_kernel = true;
if(argc == 1)
{
// use default
}
else if(argc == 3)
{
do_verification = std::stoi(argv[1]);
time_kernel = std::stoi(argv[2]);
}
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;
@@ -157,6 +174,7 @@ int main()
std::cout << "Time elapase is : " << ela_time << " ms . " << std::endl;
bool pass = true;
if(do_verification)
{
std::vector<std::size_t> mn = {static_cast<unsigned long>(M),
static_cast<unsigned long>(N)};