From ecdfe960921032c1aae6dc2c4a3e0ad1b8bba559 Mon Sep 17 00:00:00 2001 From: rocking Date: Tue, 17 May 2022 20:56:16 +0800 Subject: [PATCH] Add comment --- .../ck/tensor_operation/gpu/device/device_binary_elementwise.hpp | 1 + 1 file changed, 1 insertion(+) diff --git a/include/ck/tensor_operation/gpu/device/device_binary_elementwise.hpp b/include/ck/tensor_operation/gpu/device/device_binary_elementwise.hpp index 2cbb3f2f24..0b08b818a3 100644 --- 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)