Commit 976815e5 authored by rocking's avatar rocking
Browse files

Prevent compile error when user pass rvalue, eg {3, 4}

parent f919809d
......@@ -15,10 +15,10 @@ struct DeviceBinaryElementwise : public BaseOperator
virtual std::unique_ptr<BaseArgument> MakeArgumentPointer(const void* p_a,
const void* p_b,
void* p_c,
const std::vector<int>& shape_a,
const std::vector<int>& stride_a,
const std::vector<int>& shape_b,
const std::vector<int>& stride_b,
std::vector<int> shape_a,
std::vector<int> stride_a,
std::vector<int> shape_b,
std::vector<int> stride_b,
ElementwiseFunctor functor,
index_t threadPerBlock) = 0;
......
......@@ -163,10 +163,10 @@ struct DeviceBinaryElementwise_2D : public DeviceBinaryElementwise<ElementwiseFu
std::unique_ptr<BaseArgument> MakeArgumentPointer(const void* p_a,
const void* p_b,
void* p_c,
const std::vector<int>& shape,
const std::vector<int>& stride_a,
const std::vector<int>& stride_b,
const std::vector<int>& stride_c,
std::vector<int> shape,
std::vector<int> stride_a,
std::vector<int> stride_b,
std::vector<int> stride_c,
ElementwiseFunctor functor,
index_t threadPerBlock) override
{
......
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