Skip to content

GitLab

Commit ecdfe960 authored by rocking's avatar rocking
Browse files

Add comment

parent 492da459
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
include/ck/tensor_operation/gpu/device/device_binary_elementwise.hpp
View file @ ecdfe960
...@@ -81,6 +81,7 @@ struct DeviceBinaryElementwise : public BaseOperator ...@@ -81,6 +81,7 @@ struct DeviceBinaryElementwise : public BaseOperator
static_assert(Dim == 1 || Dim == 2, static_assert(Dim == 1 || Dim == 2,
"wrong! DeviceBinaryElementwise not support this dimension"); "wrong! DeviceBinaryElementwise not support this dimension");
// TODO - 3D, 4D, 5D
if constexpr(Dim == 1) if constexpr(Dim == 1)
return MakeDescriptor_M0_1d(shape, stride, gridSize, threadPerBlock); return MakeDescriptor_M0_1d(shape, stride, gridSize, threadPerBlock);
else if constexpr(Dim == 2) else if constexpr(Dim == 2)
......
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