Skip to content

GitLab

"include/vscode:/vscode.git/clone" did not exist on "4cdf78fc98ebe33a58f69511cfa5c934f40e43d8"
Commit cc0d4500 authored by Po-Yen, Chen's avatar Po-Yen, Chen
Browse files

Check tensor descriptor dimensions in 'GridwiseElementwise_1D'

parent b24c5f66
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
include/ck/tensor_operation/gpu/grid/gridwise_elementwise_1d.hpp
View file @ cc0d4500
...@@ -83,6 +83,8 @@ struct GridwiseElementwise_1D ...@@ -83,6 +83,8 @@ struct GridwiseElementwise_1D
auto in_global_buf_tuple = generate_tuple( auto in_global_buf_tuple = generate_tuple(
[&](auto I) { [&](auto I) {
static_assert(in_grid_1d_desc_tuple[I].GetNumOfDimension() == 1);
return make_dynamic_buffer<AddressSpaceEnum::Global>( return make_dynamic_buffer<AddressSpaceEnum::Global>(
p_in_global_tuple[I], in_grid_1d_desc_tuple[I].GetElementSpaceSize()); p_in_global_tuple[I], in_grid_1d_desc_tuple[I].GetElementSpaceSize());
}, },
...@@ -90,6 +92,8 @@ struct GridwiseElementwise_1D ...@@ -90,6 +92,8 @@ struct GridwiseElementwise_1D
auto out_global_buf_tuple = generate_tuple( auto out_global_buf_tuple = generate_tuple(
[&](auto I) { [&](auto I) {
static_assert(out_grid_1d_desc_tuple[I].GetNumOfDimension() == 1);
return make_dynamic_buffer<AddressSpaceEnum::Global>( return make_dynamic_buffer<AddressSpaceEnum::Global>(
p_out_global_tuple[I], out_grid_1d_desc_tuple[I].GetElementSpaceSize()); p_out_global_tuple[I], out_grid_1d_desc_tuple[I].GetElementSpaceSize());
}, },
......
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