We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f66832a commit 0678056Copy full SHA for 0678056
csrc/lib/ops/vec-add/torch-bind.cpp
@@ -10,8 +10,8 @@ namespace pmpp
10
{
11
12
template <typename ScalarT>
13
-auto __vector_add_impl(const torch::Tensor& A,
14
- const torch::Tensor& B) -> torch::Tensor
+auto __vector_add_impl(const torch::Tensor& A, const torch::Tensor& B)
+ -> torch::Tensor
15
16
auto nElems = pmpp::size_t(A.size(0));
17
auto C = torch::empty_like(A);
@@ -29,8 +29,8 @@ auto __vector_add_impl(const torch::Tensor& A,
29
return C;
30
}
31
32
-auto __vector_add(const torch::Tensor& A,
33
+auto __vector_add(const torch::Tensor& A, const torch::Tensor& B)
34
35
// Check if tensors are on the same device
36
TORCH_CHECK(A.device() == B.device(),
0 commit comments