Fix the padding

d7112d37 · rocking · 0e6bf342 · d7112d37
Commit d7112d37 authored Apr 20, 2022 by rocking
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

include/ck/tensor_operation/gpu/device/device_binary_elementwise_2d.hpp ...sor_operation/gpu/device/device_binary_elementwise_2d.hpp +1 -1

No files found.
--- a/include/ck/tensor_operation/gpu/device/device_binary_elementwise_2d.hpp
+++ b/include/ck/tensor_operation/gpu/device/device_binary_elementwise_2d.hpp
@@ -42,7 +42,7 @@ struct DeviceBinaryElementwise_2D : public DeviceBinaryElementwise<ElementwiseFu

        // pad
        const auto m0  = desc_m0.GetLength(I0);
-        const auto pad = math::integer_least_multiple(m0, ScalarPerVector) - m0;
+        const auto pad = math::integer_least_multiple(m0, BlockTileSize) - m0;
        const auto desc_m0_pad =
            transform_tensor_descriptor(desc_m0,
                                        make_tuple(make_right_pad_transform(m0, pad)),