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

Merge commit '4d041837ade7ae01900a0442d939f80b723b1631' into develop

Browse Source
This commit is contained in:
assistant-librarian[bot]
2025-09-03 07:12:04 +00:00
parent 1b74a93297
commit 2057f7d042
88 changed files with 21219 additions and 856 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
example/ck_tile/21_elementwise/elementwise_example.cpp
View File

@@ -5,6 +5,7 @@
#include "ck_tile/host.hpp"
#include "ck_tile/ops/elementwise.hpp"
#include "ck_tile/host/reference/reference_elementwise.hpp"
#include "json_dump.hpp"
auto create_args(int argc, char* argv[])
{
@@ -15,7 +16,9 @@ auto create_args(int argc, char* argv[])
.insert("v", "1", "cpu validation or not")
.insert("prec", "fp16", "precision")
.insert("warmup", "10", "cold iter")
.insert("repeat", "50", "hot iter");
.insert("repeat", "50", "hot iter")
.insert("json", "0", "0: No Json, 1: Dump Results in Json format")
.insert("jsonfile", "elementwise.json", "json file name to dump results");
bool result = arg_parser.parse(argc, argv);
return std::make_tuple(result, arg_parser);
@@ -195,6 +198,18 @@ bool run(const ck_tile::ArgParser& arg_parser)
y_validation, y_host, "Elementwise Add Error: Incorrect results!", 0.01, 0.01);
}
if(arg_parser.get_int("json") == 1)
{
dump_elementwise_json_results(arg_parser.get_str("jsonfile"),
arg_parser.get_str("prec"),
kGridSize,
kBlockSize,
ave_time,
0,
0,
"elementwise_add");
}
return pass;
}