Skip to content

GitLab

Unverified Commit 999c834c authored by dev-fennek's avatar dev-fennek Committed by GitHub
Browse files

Update all.cc 

[for torch 1.4]
torch::jit::RegisterOperators --> torch::RegisterOperators

ref) https://github.com/pytorch/pytorch/commit/243298668caca7ee7de454ebc4d99c487ec9b272
parent 6e727bcd
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/spconv/all.cc
View file @ 999c834c
...@@ -19,7 +19,7 @@ ...@@ -19,7 +19,7 @@
#include <spconv/nms_ops.h> #include <spconv/nms_ops.h>
static auto registry = static auto registry =
torch::jit::RegisterOperators("spconv::get_indice_pairs_2d", &spconv::getIndicePair<2>) torch::RegisterOperators("spconv::get_indice_pairs_2d", &spconv::getIndicePair<2>)
.op("spconv::get_indice_pairs_3d", &spconv::getIndicePair<3>) .op("spconv::get_indice_pairs_3d", &spconv::getIndicePair<3>)
.op("spconv::get_indice_pairs_4d", &spconv::getIndicePair<4>) .op("spconv::get_indice_pairs_4d", &spconv::getIndicePair<4>)
.op("spconv::get_indice_pairs_grid_2d", &spconv::getIndicePairPreGrid<2>) .op("spconv::get_indice_pairs_grid_2d", &spconv::getIndicePairPreGrid<2>)
......
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