wassname/Pointnet2_PyTorch
1
0
Fork 0
mirror of https://github.com/wassname/Pointnet2_PyTorch.git synced 2026-07-09 00:20:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

Initial commit

Browse Source
This commit is contained in:
erikwijmans
2017-12-26 18:43:17 -05:00
commit dc4e2b0db3
42 changed files with 3486 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/cinclude/cuda_utils.h
+24
View File
@@ -0,0 +1,24 @@
#ifndef _CUDA_UTILS_H
#define _CUDA_UTILS_H
#ifdef __cplusplus
extern "C" {
#endif
inline int opt_n_threads(int work_size) {
unsigned int n_threads = work_size;
n_threads--;
n_threads |= n_threads >> 1;
n_threads |= n_threads >> 2;
n_threads |= n_threads >> 4;
n_threads |= n_threads >> 8;
n_threads |= n_threads >> 16;
n_threads++;
return max(min(n_threads / 2, 512), 2);
}
#ifdef __cplusplus
}
#endif
#endif