support device dispatch in parrots (#1588)

a4dc2a72 · pc · GitHub · 0bcbeadb · a4dc2a72 · a4dc2a72
Unverified Commit a4dc2a72 authored Dec 24, 2021 by pc Committed by GitHub Dec 24, 2021
6 changed files
--- a/mmcv/ops/csrc/parrots/three_nn.cpp
+++ b/mmcv/ops/csrc/parrots/three_nn.cpp
--- a/mmcv/ops/csrc/parrots/tin_shift.cpp
+++ b/mmcv/ops/csrc/parrots/tin_shift.cpp
--- a/mmcv/ops/csrc/parrots/upfirdn2d.cpp
+++ b/mmcv/ops/csrc/parrots/upfirdn2d.cpp
--- a/mmcv/ops/csrc/parrots/voxelization.cpp
+++ b/mmcv/ops/csrc/parrots/voxelization.cpp
--- a/mmcv/ops/csrc/parrots/voxelization_cpu.cpp
+++ b/mmcv/ops/csrc/parrots/voxelization_cpu.cpp
--- a/setup.py
+++ b/setup.py