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
42namespace apollo {
43namespace perception {
44namespace lidar {
45
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
void DoScatterCuda(const int pillar_count, int *x_coors, int *y_coors, float *pfe_output, float *scattered_feature)
Call scatter cuda kernel
ScatterCuda(const int num_threads, const int grid_x_size, const int grid_y_size)
Constructor
class register implement
Definition arena_queue.h:37