Skip to content

Commit 0678056

Browse files
[skip ci] Auto-format code with clang-format and black
1 parent f66832a commit 0678056

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

csrc/lib/ops/vec-add/torch-bind.cpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -10,8 +10,8 @@ namespace pmpp
1010
{
1111

1212
template <typename ScalarT>
13-
auto __vector_add_impl(const torch::Tensor& A,
14-
const torch::Tensor& B) -> torch::Tensor
13+
auto __vector_add_impl(const torch::Tensor& A, const torch::Tensor& B)
14+
-> torch::Tensor
1515
{
1616
auto nElems = pmpp::size_t(A.size(0));
1717
auto C = torch::empty_like(A);
@@ -29,8 +29,8 @@ auto __vector_add_impl(const torch::Tensor& A,
2929
return C;
3030
}
3131

32-
auto __vector_add(const torch::Tensor& A,
33-
const torch::Tensor& B) -> torch::Tensor
32+
auto __vector_add(const torch::Tensor& A, const torch::Tensor& B)
33+
-> torch::Tensor
3434
{
3535
// Check if tensors are on the same device
3636
TORCH_CHECK(A.device() == B.device(),

0 commit comments

Comments
 (0)