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
......@@ -19,7 +19,7 @@
#include <spconv/nms_ops.h>
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_4d", &spconv::getIndicePair<4>)
.op("spconv::get_indice_pairs_grid_2d", &spconv::getIndicePairPreGrid<2>)
......@@ -39,4 +39,4 @@ static auto registry =
&spconv::indiceMaxPoolBackward<at::Half>)
.op("spconv::nms", &spconv::nonMaxSuppression<float>)
.op("spconv::pillar_scatter_float", &spconv::pointPillarScatter<float>)
.op("spconv::pillar_scatter_half", &spconv::pointPillarScatter<at::Half>);
\ No newline at end of file
.op("spconv::pillar_scatter_half", &spconv::pointPillarScatter<at::Half>);
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