Cheetah - SKA - PSS - Prototype Time Domain Search Pipeline
Rfim.cu
1 /*
2  * The MIT License (MIT)
3  *
4  * Copyright (c) 2016 The SKA organisation
5  *
6  * Permission is hereby granted, free of charge, to any person obtaining a copy
7  * of this software and associated documentation files (the "Software"), to deal
8  * in the Software without restriction, including without limitation the rights
9  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10  * copies of the Software, and to permit persons to whom the Software is
11  * furnished to do so, subject to the following conditions:
12  *
13  * The above copyright notice and this permission notice shall be included in all
14  * copies or substantial portions of the Software.
15  *
16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22  * SOFTWARE.
23  */
24 // For the GPU we use char instead of bool
25 typedef unsigned char LocalFlagsType;
26 
27 #include "cheetah/rfim/cuda/src/Kernels.cu"
28 #include "cheetah/rfim/cuda/Rfim.h"
29 #include "panda/arch/nvidia/CudaDevicePointer.h"
30 #include "panda/Error.h"
31 
32 
33 namespace ska {
34 namespace cheetah {
35 namespace rfim {
36 namespace cuda {
37 
38 template<typename TimeFrequencyDataType>
39 void flagger_freq_kernel_template(TimeFrequencyDataType const& data, unsigned char* host_flags)
40 {
41  typedef typename std::decay<decltype(data)>::type::DataType DataType;
42 
43  std::size_t data_size=data.number_of_channels() * data.number_of_spectra();
44 
45  panda::nvidia::CudaDevicePointer<DataType> device_data(data_size);
46  panda::nvidia::CudaDevicePointer<LocalFlagsType> device_flags(data_size);
47  panda::nvidia::CudaDevicePointer<unsigned> device_nr_flagged(data.number_of_spectra());
48 
49  // Copy input data to device
50  device_data.write(data.begin(), data.end());
51 
52  // Execute code on the device
53  // Frequency Flagger
54  flagger_freq<<<data.number_of_spectra(), data.number_of_channels(), (data.number_of_channels() * sizeof(DataType)) + (data.number_of_channels() * sizeof(LocalFlagsType))>>>(device_data(), device_flags(), device_nr_flagged(), data.number_of_channels(), data.number_of_spectra());
55 
56  // Copy flags data to host
57  device_flags.read(host_flags);
58 
59 }
60 
61 void flagger_freq_kernel(data::TimeFrequency<Cpu, uint8_t> const& data, unsigned char* host_flags)
62 {
63  flagger_freq_kernel_template(data, host_flags);
64 }
65 
66 } // namespace cuda
67 } // namespace rfim
68 } // namespace cheetah
69 } // namespace ska
Some limits and constants for FLDO.
Definition: Brdz.h:35