Skip to content

Commit 33f2937

Browse files
[skip ci] Auto-format code with clang-format and black
1 parent 6fbfaaf commit 33f2937

File tree

2 files changed

+5
-9
lines changed

2 files changed

+5
-9
lines changed

csrc/lib/ops/torch_impl.hpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,8 @@
55
namespace pmpp::ops::cpu::torch_impl
66
{
77

8-
auto vectorAdd(const torch::Tensor& A,
9-
const torch::Tensor& B) -> torch::Tensor;
8+
auto vectorAdd(const torch::Tensor& A, const torch::Tensor& B)
9+
-> torch::Tensor;
1010

1111
auto cvtRGBtoGray(const torch::Tensor& img) -> torch::Tensor;
1212

@@ -17,8 +17,8 @@ auto matmul(const torch::Tensor& A, const torch::Tensor& B) -> torch::Tensor;
1717
namespace pmpp::ops::cuda::torch_impl
1818
{
1919

20-
auto vectorAdd(const torch::Tensor& A,
21-
const torch::Tensor& B) -> torch::Tensor;
20+
auto vectorAdd(const torch::Tensor& A, const torch::Tensor& B)
21+
-> torch::Tensor;
2222

2323
auto cvtRGBtoGray(const torch::Tensor& img) -> torch::Tensor;
2424

test/test.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -19,8 +19,4 @@
1919
pic_out_cuda = torch.ops.pmpp.cvt_rgb_to_gray(pic_in)
2020
print(pic_out_cuda.cpu())
2121

22-
print(
23-
torch.ops.pmpp.matmul(
24-
torch.ones((32, 32)).cuda(), torch.ones((32, 32)).cuda()
25-
)
26-
)
22+
print(torch.ops.pmpp.matmul(torch.ones((32, 32)).cuda(), torch.ones((32, 32)).cuda()))

0 commit comments

Comments
 (0)