Commit b256a3b0 authored by Paul's avatar Paul
Browse files

Format

parent 96e062d1
...@@ -42,9 +42,9 @@ struct hip_compile_options ...@@ -42,9 +42,9 @@ struct hip_compile_options
std::size_t local; std::size_t local;
std::vector<shape> inputs; std::vector<shape> inputs;
shape output; shape output;
std::string kernel_name = "kernel"; std::string kernel_name = "kernel";
std::string params = ""; std::string params = "";
std::vector<shape> virtual_inputs = {}; std::vector<shape> virtual_inputs = {};
std::vector<src_file> additional_src_files = {}; std::vector<src_file> additional_src_files = {};
/** /**
......
...@@ -291,9 +291,12 @@ struct ck_gemm_compiler : compiler<ck_gemm_compiler> ...@@ -291,9 +291,12 @@ struct ck_gemm_compiler : compiler<ck_gemm_compiler>
const auto block_size = solution.block_size; const auto block_size = solution.block_size;
hip_compile_options options; hip_compile_options options;
std::transform(ck_headers.begin(), ck_headers.end(), std::back_inserter(options.additional_src_files), [&](auto&& p) { std::transform(ck_headers.begin(),
return src_file{fs::path{p.first}, p.second}; 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; 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.set_launch_params(v, grid_size * block_size, block_size);
options.inputs = inputs; 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