Skip to content
Snippets Groups Projects
Unverified Commit 57283a76 authored by Gu Wang's avatar Gu Wang Committed by GitHub
Browse files

Merge pull request #81 from belalhmedan90/main

remove dependency to THC
parents f3ca1863 9bdbfb6e
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
* 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)
* `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
......
#include <torch/extension.h>
#include <ATen/cuda/CUDAContext.h>
#include <THC/THC.h>
#include <THC/THCAtomics.cuh>
#include <THC/THCDeviceUtils.cuh>
#include <cfloat>
#include <cuda.h>
......
......@@ -3,8 +3,6 @@
#include <iostream>
#include <vector>
extern THCState* state;
#define CHECK_CUDA(x) \
do { \
......
File deleted
File deleted
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