Add comment

ecdfe960 · rocking · 492da459 · ecdfe960
Commit ecdfe960 authored May 17, 2022 by rocking
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

include/ck/tensor_operation/gpu/device/device_binary_elementwise.hpp ...tensor_operation/gpu/device/device_binary_elementwise.hpp +1 -0

No files found.
--- a/include/ck/tensor_operation/gpu/device/device_binary_elementwise.hpp
+++ b/include/ck/tensor_operation/gpu/device/device_binary_elementwise.hpp
@@ -81,6 +81,7 @@ struct DeviceBinaryElementwise : public BaseOperator
        static_assert(Dim == 1 || Dim == 2,
                      "wrong! DeviceBinaryElementwise not support this dimension");

+        // TODO - 3D, 4D, 5D
        if constexpr(Dim == 1)
            return MakeDescriptor_M0_1d(shape, stride, gridSize, threadPerBlock);
        else if constexpr(Dim == 2)