Check tensor descriptor dimensions in 'GridwiseElementwise_1D'

cc0d4500 · Po-Yen, Chen · b24c5f66 · cc0d4500
Commit cc0d4500 authored Sep 07, 2022 by Po-Yen, Chen
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

include/ck/tensor_operation/gpu/grid/gridwise_elementwise_1d.hpp .../ck/tensor_operation/gpu/grid/gridwise_elementwise_1d.hpp +4 -0

No files found.
--- a/include/ck/tensor_operation/gpu/grid/gridwise_elementwise_1d.hpp
+++ b/include/ck/tensor_operation/gpu/grid/gridwise_elementwise_1d.hpp
@@ -83,6 +83,8 @@ struct GridwiseElementwise_1D

        auto in_global_buf_tuple = generate_tuple(
            [&](auto I) {
+                static_assert(in_grid_1d_desc_tuple[I].GetNumOfDimension() == 1);
+
                return make_dynamic_buffer<AddressSpaceEnum::Global>(
                    p_in_global_tuple[I], in_grid_1d_desc_tuple[I].GetElementSpaceSize());
            },
@@ -90,6 +92,8 @@ struct GridwiseElementwise_1D

        auto out_global_buf_tuple = generate_tuple(
            [&](auto I) {
+                static_assert(out_grid_1d_desc_tuple[I].GetNumOfDimension() == 1);
+
                return make_dynamic_buffer<AddressSpaceEnum::Global>(
                    p_out_global_tuple[I], out_grid_1d_desc_tuple[I].GetElementSpaceSize());
            },