Skip to content

Commit 6c8570c

Browse files
committed
refactor: rename scan_cpu and lpc_cpu functions to scan and lpc
1 parent 4dcde47 commit 6c8570c

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

torchlpc/csrc/scan_cpu.cpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -142,11 +142,11 @@ at::Tensor lpc_cpu(const at::Tensor &x, const at::Tensor &a,
142142
}
143143

144144
TORCH_LIBRARY(torchlpc, m) {
145-
m.def("torchlpc::scan_cpu(Tensor a, Tensor b, Tensor c) -> Tensor");
146-
m.def("torchlpc::lpc_cpu(Tensor a, Tensor b, Tensor c) -> Tensor");
145+
m.def("torchlpc::scan(Tensor a, Tensor b, Tensor c) -> Tensor");
146+
m.def("torchlpc::lpc(Tensor a, Tensor b, Tensor c) -> Tensor");
147147
}
148148

149149
TORCH_LIBRARY_IMPL(torchlpc, CPU, m) {
150-
m.impl("scan_cpu", &scan_cpu_wrapper);
151-
m.impl("lpc_cpu", &lpc_cpu);
150+
m.impl("scan", &scan_cpu_wrapper);
151+
m.impl("lpc", &lpc_cpu);
152152
}

0 commit comments

Comments
 (0)