Skip to content

GitLab

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

fixed NumDim dimension error

parent 4dfcf974
Show 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 {
template <typename InDataTypeTuple,
typename OutDataTypeTuple,
typename ElementwiseOperation,
index_t NumDim,
index_t NumDim_m,
index_t NumDim_n,
index_t MPerThread,
index_t NPerThread,
typename InScalarPerVectorSeq,
typename OutScalarPerVectorSeq>
struct DeviceElementwise : public DeviceElementwiseBase<InDataTypeTuple,
OutDataTypeTuple,
ElementwiseOperation,
NumDim_m,
NumDim_n>
struct DeviceElementwise
: public DeviceElementwiseBase<InDataTypeTuple, OutDataTypeTuple, ElementwiseOperation, NumDim>
{
static constexpr int NumInput = InDataTypeTuple::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() &&
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