Commit 641938fd authored by zhanggzh's avatar zhanggzh
Browse files

add mmdetection3d source code and change readme

parent fe25f7a5
This diff is collapsed.
...@@ -13,7 +13,7 @@ typedef enum { SUM = 0, MEAN = 1, MAX = 2 } reduce_t; ...@@ -13,7 +13,7 @@ typedef enum { SUM = 0, MEAN = 1, MAX = 2 } reduce_t;
#define CHECK_INPUT(x) \ #define CHECK_INPUT(x) \
CHECK_CUDA(x); \ CHECK_CUDA(x); \
CHECK_CONTIGUOUS(x) CHECK_CONTIGUOUS(x)
#define __CUDA_ARCH__ 750
namespace { namespace {
int const threadsPerBlock = 512; int const threadsPerBlock = 512;
int const maxGridDim = 50000; int const maxGridDim = 50000;
......
...@@ -50,19 +50,19 @@ if __name__ == '__main__': ...@@ -50,19 +50,19 @@ if __name__ == '__main__':
name='bev_pool_ext', name='bev_pool_ext',
module='projects.BEVFusion.bevfusion.ops.bev_pool', module='projects.BEVFusion.bevfusion.ops.bev_pool',
sources=[ sources=[
'src/bev_pool.cpp', '/home/mmdetection3d/projects/BEVFusion/bevfusion/ops/bev_pool/src/bev_pool.cpp',
'src/bev_pool_cuda.cu', '/home/mmdetection3d/projects/BEVFusion/bevfusion/ops/bev_pool/src/bev_pool_cuda.cu',
], ],
), ),
make_cuda_ext( make_cuda_ext(
name='voxel_layer', name='voxel_layer',
module='projects.BEVFusion.bevfusion.ops.voxel', module='projects.BEVFusion.bevfusion.ops.voxel',
sources=[ sources=[
'src/voxelization.cpp', '/home/mmdetection3d/projects/BEVFusion/bevfusion/ops/voxel/src/voxelization.cpp',
'src/scatter_points_cpu.cpp', '/home/mmdetection3d/projects/BEVFusion/bevfusion/ops/voxel/src/scatter_points_cpu.cpp',
'src/scatter_points_cuda.cu', '/home/mmdetection3d/projects/BEVFusion/bevfusion/ops/voxel/src/scatter_points_cuda.cu',
'src/voxelization_cpu.cpp', '/home/mmdetection3d/projects/BEVFusion/bevfusion/ops/voxel/src/voxelization_cpu.cpp',
'src/voxelization_cuda.cu', '/home/mmdetection3d/projects/BEVFusion/bevfusion/ops/voxel/src/voxelization_cuda.cu',
], ],
), ),
], ],
......
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