Skip to content

GitLab

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

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

parent f919809d
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 8 deletions
include/ck/tensor_operation/gpu/device/device_binary_elementwise.hpp
View file @ 976815e5
...@@ -15,10 +15,10 @@ struct DeviceBinaryElementwise : public BaseOperator ...@@ -15,10 +15,10 @@ struct DeviceBinaryElementwise : public BaseOperator
virtual std::unique_ptr<BaseArgument> MakeArgumentPointer(const void* p_a, virtual std::unique_ptr<BaseArgument> MakeArgumentPointer(const void* p_a,
const void* p_b, const void* p_b,
void* p_c, void* p_c,
const std::vector<int>& shape_a, std::vector<int> shape_a,
const std::vector<int>& stride_a, std::vector<int> stride_a,
const std::vector<int>& shape_b, std::vector<int> shape_b,
const std::vector<int>& stride_b, std::vector<int> stride_b,
ElementwiseFunctor functor, ElementwiseFunctor functor,
index_t threadPerBlock) = 0; index_t threadPerBlock) = 0;
......
include/ck/tensor_operation/gpu/device/device_binary_elementwise_2d.hpp
View file @ 976815e5
...@@ -163,10 +163,10 @@ struct DeviceBinaryElementwise_2D : public DeviceBinaryElementwise<ElementwiseFu ...@@ -163,10 +163,10 @@ struct DeviceBinaryElementwise_2D : public DeviceBinaryElementwise<ElementwiseFu
std::unique_ptr<BaseArgument> MakeArgumentPointer(const void* p_a, std::unique_ptr<BaseArgument> MakeArgumentPointer(const void* p_a,
const void* p_b, const void* p_b,
void* p_c, void* p_c,
const std::vector<int>& shape, std::vector<int> shape,
const std::vector<int>& stride_a, std::vector<int> stride_a,
const std::vector<int>& stride_b, std::vector<int> stride_b,
const std::vector<int>& stride_c, std::vector<int> stride_c,
ElementwiseFunctor functor, ElementwiseFunctor functor,
index_t threadPerBlock) override 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