Commit b256a3b0 authored by Paul's avatar Paul
Browse files

Format

parent 96e062d1
......@@ -42,9 +42,9 @@ struct hip_compile_options
std::size_t local;
std::vector<shape> inputs;
shape output;
std::string kernel_name = "kernel";
std::string params = "";
std::vector<shape> virtual_inputs = {};
std::string kernel_name = "kernel";
std::string params = "";
std::vector<shape> virtual_inputs = {};
std::vector<src_file> additional_src_files = {};
/**
......
......@@ -291,9 +291,12 @@ struct ck_gemm_compiler : compiler<ck_gemm_compiler>
const auto block_size = solution.block_size;
hip_compile_options options;
std::transform(ck_headers.begin(), ck_headers.end(), std::back_inserter(options.additional_src_files), [&](auto&& p) {
return src_file{fs::path{p.first}, p.second};
});
std::transform(ck_headers.begin(),
ck_headers.end(),
std::back_inserter(options.additional_src_files),
[&](auto&& p) {
return src_file{fs::path{p.first}, p.second};
});
auto grid_size = can_fold_batch ? blocks_per_batch : batch_count * blocks_per_batch;
options.set_launch_params(v, grid_size * block_size, block_size);
options.inputs = inputs;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment