densedense_to_sparse.hpp

//*LB*
// Copyright (c) 2010, University of Bonn, Institute for Computer Science VI
// 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 University of Bonn 
//    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 THE COPYRIGHT HOLDER OR CONTRIBUTORS 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.
//*LE*





#ifndef __DENSEDENSE_TO_SPARSE_HPP__
#define __DENSEDENSE_TO_SPARSE_HPP__


#include <cuv/basics/dia_matrix.hpp>

#define SPARSE_DIA_BLOCK_SIZE 16
#define SPARSE_DIA_BLOCK_SIZE_LEN (2*SPARSE_DIA_BLOCK_SIZE+2)

namespace cuv{

        template<class __value_type, class __index_type=unsigned int>
        class host_block_descriptor{
                public:
                        typedef __value_type value_type;  
                        typedef __index_type index_type;  
                        typedef dia_matrix<value_type,host_memory_space,index_type> diamat_type; 

                        host_block_descriptor(const diamat_type& d){}
        };

        template<class __value_type, class __index_type=unsigned int>
        class dev_block_descriptor{
                public:
                        typedef __value_type value_type; 
                        typedef __index_type index_type; 
                        typedef dia_matrix<value_type,dev_memory_space,index_type> diamat_type; 
                protected:
                        struct block{
                                int              startx;  
                                int                              starty;  
                                int              diag[2*SPARSE_DIA_BLOCK_SIZE];  
                        };
                        struct block_array{  
                                int*    ptr;     
                                int     len;     
                        } m_blocks;          
                public:
                        dev_block_descriptor(const diamat_type& d);
                        ~dev_block_descriptor();

                        const block_array& blocks()const{return m_blocks;}

                        inline int len()const{ return m_blocks.len; }
        };

        // dev_block_descriptor and host_block_descriptor explicit... UGLY...
        template<class __value_type, class __memory_layout_type, class __index_type >
        void densedense_to_dia(
                   dia_matrix<__value_type,dev_memory_space,__index_type>&           C,
                   const dev_block_descriptor<__value_type, __index_type>&      Cbd,
                   const tensor<__value_type, dev_memory_space, __memory_layout_type>&   A,
                   const tensor<__value_type, dev_memory_space, __memory_layout_type>&   B,
                   const __value_type& factAB=1.f,
                   const __value_type& factC =0.f);

        template<class __value_type, class __memory_layout_type, class __index_type >
        void densedense_to_dia(
                   dia_matrix<__value_type,host_memory_space,__index_type>&           C,
                   const host_block_descriptor<__value_type, __index_type>&      Cbd,
                   const tensor<__value_type, host_memory_space, __memory_layout_type>&   A,
                   const tensor<__value_type, host_memory_space, __memory_layout_type>&   B,
                   const __value_type& factAB=1.f,
                   const __value_type& factC =0.f);

        
} // cuv


#endif