Skip to content

GitLab

Commit 76b44c60 authored by Astha Rai's avatar Astha Rai
Browse files

fixed NumDim dimension error

parent 4dfcf974
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 6 deletions
include/ck/tensor_operation/gpu/device/device_elementwise_2d.hpp
View file @ 76b44c60
...@@ -20,22 +20,20 @@ namespace device { ...@@ -20,22 +20,20 @@ namespace device {
template <typename InDataTypeTuple, template <typename InDataTypeTuple,
typename OutDataTypeTuple, typename OutDataTypeTuple,
typename ElementwiseOperation, typename ElementwiseOperation,
index_t NumDim,
index_t NumDim_m, index_t NumDim_m,
index_t NumDim_n, index_t NumDim_n,
index_t MPerThread, index_t MPerThread,
index_t NPerThread, index_t NPerThread,
typename InScalarPerVectorSeq, typename InScalarPerVectorSeq,
typename OutScalarPerVectorSeq> typename OutScalarPerVectorSeq>
struct DeviceElementwise : public DeviceElementwiseBase<InDataTypeTuple, struct DeviceElementwise
OutDataTypeTuple, : public DeviceElementwiseBase<InDataTypeTuple, OutDataTypeTuple, ElementwiseOperation, NumDim>
ElementwiseOperation,
NumDim_m,
NumDim_n>
{ {
static constexpr int NumInput = InDataTypeTuple::Size(); static constexpr int NumInput = InDataTypeTuple::Size();
static constexpr int NumOutput = OutDataTypeTuple::Size(); static constexpr int NumOutput = OutDataTypeTuple::Size();
const index_t NumDim = NumDim_m + NumDim_n; // const index_t NumDim = NumDim_m + NumDim_n;
static_assert(NumInput == InScalarPerVectorSeq::Size() && static_assert(NumInput == InScalarPerVectorSeq::Size() &&
NumOutput == OutScalarPerVectorSeq::Size(), NumOutput == OutScalarPerVectorSeq::Size(),
......
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