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;
#define CHECK_INPUT(x) \
CHECK_CUDA(x); \
CHECK_CONTIGUOUS(x)
#define __CUDA_ARCH__ 750
namespace {
int const threadsPerBlock = 512;
int const maxGridDim = 50000;
......
......@@ -50,19 +50,19 @@ if __name__ == '__main__':
name='bev_pool_ext',
module='projects.BEVFusion.bevfusion.ops.bev_pool',
sources=[
'src/bev_pool.cpp',
'src/bev_pool_cuda.cu',
'/home/mmdetection3d/projects/BEVFusion/bevfusion/ops/bev_pool/src/bev_pool.cpp',
'/home/mmdetection3d/projects/BEVFusion/bevfusion/ops/bev_pool/src/bev_pool_cuda.cu',
],
),
make_cuda_ext(
name='voxel_layer',
module='projects.BEVFusion.bevfusion.ops.voxel',
sources=[
'src/voxelization.cpp',
'src/scatter_points_cpu.cpp',
'src/scatter_points_cuda.cu',
'src/voxelization_cpu.cpp',
'src/voxelization_cuda.cu',
'/home/mmdetection3d/projects/BEVFusion/bevfusion/ops/voxel/src/voxelization.cpp',
'/home/mmdetection3d/projects/BEVFusion/bevfusion/ops/voxel/src/scatter_points_cpu.cpp',
'/home/mmdetection3d/projects/BEVFusion/bevfusion/ops/voxel/src/scatter_points_cuda.cu',
'/home/mmdetection3d/projects/BEVFusion/bevfusion/ops/voxel/src/voxelization_cpu.cpp',
'/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