Apollo
11.0
自动驾驶开放平台
scatter_cuda.h
浏览该文件的文档.
1
/******************************************************************************
2
* Copyright 2020 The Apollo Authors. All Rights Reserved.
3
*
4
* Licensed under the Apache License, Version 2.0 (the "License");
5
* you may not use this file except in compliance with the License.
6
* You may obtain a copy of the License at
7
*
8
* http://www.apache.org/licenses/LICENSE-2.0
9
*
10
* Unless required by applicable law or agreed to in writing, software
11
* distributed under the License is distributed on an "AS IS" BASIS,
12
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13
* See the License for the specific language governing permissions and
14
* limitations under the License.
15
*****************************************************************************/
16
17
/*
18
* Copyright 2018-2019 Autoware Foundation. All rights reserved.
19
*
20
* Licensed under the Apache License, Version 2.0 (the "License");
21
* you may not use this file except in compliance with the License.
22
* You may obtain a copy of the License at
23
*
24
* http://www.apache.org/licenses/LICENSE-2.0
25
*
26
* Unless required by applicable law or agreed to in writing, software
27
* distributed under the License is distributed on an "AS IS" BASIS,
28
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
29
* See the License for the specific language governing permissions and
30
* limitations under the License.
31
*/
32
40
#pragma once
41
42
namespace
apollo
{
43
namespace
perception {
44
namespace
lidar {
45
46
class
ScatterCuda
{
47
private
:
48
const
int
num_threads_;
49
const
int
grid_x_size_;
50
const
int
grid_y_size_;
51
52
public
:
60
ScatterCuda
(
const
int
num_threads,
const
int
grid_x_size,
61
const
int
grid_y_size);
62
73
void
DoScatterCuda
(
const
int
pillar_count,
int
* x_coors,
int
* y_coors,
74
float
* pfe_output,
float
* scattered_feature);
75
};
76
77
}
// namespace lidar
78
}
// namespace perception
79
}
// namespace apollo
apollo::perception::lidar::ScatterCuda
Definition
scatter_cuda.h:46
apollo::perception::lidar::ScatterCuda::DoScatterCuda
void DoScatterCuda(const int pillar_count, int *x_coors, int *y_coors, float *pfe_output, float *scattered_feature)
Call scatter cuda kernel
apollo::perception::lidar::ScatterCuda::ScatterCuda
ScatterCuda(const int num_threads, const int grid_x_size, const int grid_y_size)
Constructor
apollo
class register implement
Definition
arena_queue.h:37
modules
perception
lidar_detection
detector
point_pillars_detection
scatter_cuda.h