/****************************************************************************** * Copyright (c) 2010-2011, Duane Merrill. All rights reserved. * Copyright (c) 2011-2018, NVIDIA CORPORATION. All rights reserved. * Modifications Copyright (c) 2017-2020, Advanced Micro Devices, Inc. All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions are met: * * Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * * Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * * Neither the name of the NVIDIA CORPORATION nor the * names of its contributors may be used to endorse or promote products * derived from this software without specific prior written permission. * * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE * DISCLAIMED. IN NO EVENT SHALL NVIDIA CORPORATION BE LIABLE FOR ANY * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. * ******************************************************************************/ #ifndef HIPCUB_ROCPRIM_DEVICE_DEVICE_HISTOGRAM_HPP_ #define HIPCUB_ROCPRIM_DEVICE_DEVICE_HISTOGRAM_HPP_ #include "../config.hpp" #include "../util_type.cuh" #include BEGIN_HIPCUB_NAMESPACE struct DeviceHistogram { template< typename SampleIteratorT, typename CounterT, typename LevelT, typename OffsetT > HIPCUB_RUNTIME_FUNCTION static cudaError_t HistogramEven(void * d_temp_storage, size_t& temp_storage_bytes, SampleIteratorT d_samples, CounterT * d_histogram, int num_levels, LevelT lower_level, LevelT upper_level, OffsetT num_samples, cudaStream_t stream = 0, bool debug_synchronous = false) { return (cudaError_t)::rocprim::histogram_even( d_temp_storage, temp_storage_bytes, d_samples, num_samples, d_histogram, num_levels, lower_level, upper_level, stream, debug_synchronous ); } template< typename SampleIteratorT, typename CounterT, typename LevelT, typename OffsetT > HIPCUB_RUNTIME_FUNCTION static cudaError_t HistogramEven(void * d_temp_storage, size_t& temp_storage_bytes, SampleIteratorT d_samples, CounterT * d_histogram, int num_levels, LevelT lower_level, LevelT upper_level, OffsetT num_row_samples, OffsetT num_rows, size_t row_stride_bytes, cudaStream_t stream = 0, bool debug_synchronous = false) { return (cudaError_t)::rocprim::histogram_even( d_temp_storage, temp_storage_bytes, d_samples, num_row_samples, num_rows, row_stride_bytes, d_histogram, num_levels, lower_level, upper_level, stream, debug_synchronous ); } template< int NUM_CHANNELS, int NUM_ACTIVE_CHANNELS, typename SampleIteratorT, typename CounterT, typename LevelT, typename OffsetT > HIPCUB_RUNTIME_FUNCTION static cudaError_t MultiHistogramEven(void * d_temp_storage, size_t& temp_storage_bytes, SampleIteratorT d_samples, CounterT * d_histogram[NUM_ACTIVE_CHANNELS], int num_levels[NUM_ACTIVE_CHANNELS], LevelT lower_level[NUM_ACTIVE_CHANNELS], LevelT upper_level[NUM_ACTIVE_CHANNELS], OffsetT num_pixels, cudaStream_t stream = 0, bool debug_synchronous = false) { unsigned int levels[NUM_ACTIVE_CHANNELS]; for(unsigned int channel = 0; channel < NUM_ACTIVE_CHANNELS; channel++) { levels[channel] = num_levels[channel]; } return (cudaError_t)::rocprim::multi_histogram_even( d_temp_storage, temp_storage_bytes, d_samples, num_pixels, d_histogram, levels, lower_level, upper_level, stream, debug_synchronous ); } template< int NUM_CHANNELS, int NUM_ACTIVE_CHANNELS, typename SampleIteratorT, typename CounterT, typename LevelT, typename OffsetT > HIPCUB_RUNTIME_FUNCTION static cudaError_t MultiHistogramEven(void * d_temp_storage, size_t& temp_storage_bytes, SampleIteratorT d_samples, CounterT * d_histogram[NUM_ACTIVE_CHANNELS], int num_levels[NUM_ACTIVE_CHANNELS], LevelT lower_level[NUM_ACTIVE_CHANNELS], LevelT upper_level[NUM_ACTIVE_CHANNELS], OffsetT num_row_pixels, OffsetT num_rows, size_t row_stride_bytes, cudaStream_t stream = 0, bool debug_synchronous = false) { unsigned int levels[NUM_ACTIVE_CHANNELS]; for(unsigned int channel = 0; channel < NUM_ACTIVE_CHANNELS; channel++) { levels[channel] = num_levels[channel]; } return (cudaError_t)::rocprim::multi_histogram_even( d_temp_storage, temp_storage_bytes, d_samples, num_row_pixels, num_rows, row_stride_bytes, d_histogram, levels, lower_level, upper_level, stream, debug_synchronous ); } template< typename SampleIteratorT, typename CounterT, typename LevelT, typename OffsetT > HIPCUB_RUNTIME_FUNCTION static cudaError_t HistogramRange(void * d_temp_storage, size_t& temp_storage_bytes, SampleIteratorT d_samples, CounterT * d_histogram, int num_levels, LevelT * d_levels, OffsetT num_samples, cudaStream_t stream = 0, bool debug_synchronous = false) { return (cudaError_t)::rocprim::histogram_range( d_temp_storage, temp_storage_bytes, d_samples, num_samples, d_histogram, num_levels, d_levels, stream, debug_synchronous ); } template< typename SampleIteratorT, typename CounterT, typename LevelT, typename OffsetT > HIPCUB_RUNTIME_FUNCTION static cudaError_t HistogramRange(void * d_temp_storage, size_t& temp_storage_bytes, SampleIteratorT d_samples, CounterT * d_histogram, int num_levels, LevelT * d_levels, OffsetT num_row_samples, OffsetT num_rows, size_t row_stride_bytes, cudaStream_t stream = 0, bool debug_synchronous = false) { return (cudaError_t)::rocprim::histogram_range( d_temp_storage, temp_storage_bytes, d_samples, num_row_samples, num_rows, row_stride_bytes, d_histogram, num_levels, d_levels, stream, debug_synchronous ); } template< int NUM_CHANNELS, int NUM_ACTIVE_CHANNELS, typename SampleIteratorT, typename CounterT, typename LevelT, typename OffsetT > HIPCUB_RUNTIME_FUNCTION static cudaError_t MultiHistogramRange(void * d_temp_storage, size_t& temp_storage_bytes, SampleIteratorT d_samples, CounterT * d_histogram[NUM_ACTIVE_CHANNELS], int num_levels[NUM_ACTIVE_CHANNELS], LevelT * d_levels[NUM_ACTIVE_CHANNELS], OffsetT num_pixels, cudaStream_t stream = 0, bool debug_synchronous = false) { unsigned int levels[NUM_ACTIVE_CHANNELS]; for(unsigned int channel = 0; channel < NUM_ACTIVE_CHANNELS; channel++) { levels[channel] = num_levels[channel]; } return (cudaError_t)::rocprim::multi_histogram_range( d_temp_storage, temp_storage_bytes, d_samples, num_pixels, d_histogram, levels, d_levels, stream, debug_synchronous ); } template< int NUM_CHANNELS, int NUM_ACTIVE_CHANNELS, typename SampleIteratorT, typename CounterT, typename LevelT, typename OffsetT > HIPCUB_RUNTIME_FUNCTION static cudaError_t MultiHistogramRange(void * d_temp_storage, size_t& temp_storage_bytes, SampleIteratorT d_samples, CounterT * d_histogram[NUM_ACTIVE_CHANNELS], int num_levels[NUM_ACTIVE_CHANNELS], LevelT * d_levels[NUM_ACTIVE_CHANNELS], OffsetT num_row_pixels, OffsetT num_rows, size_t row_stride_bytes, cudaStream_t stream = 0, bool debug_synchronous = false) { unsigned int levels[NUM_ACTIVE_CHANNELS]; for(unsigned int channel = 0; channel < NUM_ACTIVE_CHANNELS; channel++) { levels[channel] = num_levels[channel]; } return (cudaError_t)::rocprim::multi_histogram_range( d_temp_storage, temp_storage_bytes, d_samples, num_row_pixels, num_rows, row_stride_bytes, d_histogram, levels, d_levels, stream, debug_synchronous ); } }; END_HIPCUB_NAMESPACE #endif // HIPCUB_ROCPRIM_DEVICE_DEVICE_HISTOGRAM_HPP_