Skip to content
Snippets Groups Projects
Commit d2130681 authored by belalhmedan's avatar belalhmedan
Browse files

remove dependency to THC

parent f3ca1863
No related branches found
No related tags found
No related merge requests found
...@@ -36,9 +36,16 @@ Download the trained models at [Onedrive](https://mailstsinghuaeducn-my.sharepoi ...@@ -36,9 +36,16 @@ Download the trained models at [Onedrive](https://mailstsinghuaeducn-my.sharepoi
* Ubuntu 18.04/20.04, CUDA 10.1/10.2/11.6, python >= 3.7, PyTorch >= 1.9, torchvision * Ubuntu 18.04/20.04, CUDA 10.1/10.2/11.6, python >= 3.7, PyTorch >= 1.9, torchvision
* Install `detectron2` from [source](https://github.com/facebookresearch/detectron2) * Install `detectron2` from [source](https://github.com/facebookresearch/detectron2)
* `sh scripts/install_deps.sh` * `sh scripts/install_deps.sh`
* Compile the cpp extension for `farthest points sampling (fps)`: * Compile the cpp extensions for
1. `farthest points sampling (fps)`
2. `flow`
3. `uncertainty pnp`
4. `ransac_voting`
5. `chamfer distance`
6. `egl renderer`
``` ```
sh core/csrc/compile.sh sh ./scripts/compile_all.sh
``` ```
## Detection ## Detection
......
#include <torch/extension.h> #include <torch/extension.h>
#include <ATen/cuda/CUDAContext.h> #include <ATen/cuda/CUDAContext.h>
#include <THC/THC.h>
#include <THC/THCAtomics.cuh>
#include <THC/THCDeviceUtils.cuh>
#include <cfloat> #include <cfloat>
#include <cuda.h> #include <cuda.h>
......
...@@ -3,8 +3,6 @@ ...@@ -3,8 +3,6 @@
#include <iostream> #include <iostream>
#include <vector> #include <vector>
extern THCState* state;
#define CHECK_CUDA(x) \ #define CHECK_CUDA(x) \
do { \ do { \
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment