dispatch.hpp

Go to the documentation of this file.

#ifndef TATAMI_MULT_DENSE_MATRIX_DENSE_ROW_DISPATCH_HPP
#define TATAMI_MULT_DENSE_MATRIX_DENSE_ROW_DISPATCH_HPP
 
#include "row_to_row.hpp"
#include "row_to_column.hpp"
#include "column_to_row.hpp"
#include "column_to_column.hpp"
 
#include "../../utils.hpp"
 
namespace tatami_mult {
 
struct MultiplyDenseRowWithDenseMatrixOptions {
    MultiplyDenseRowWithDenseColumnMatrixToColumnOutputOptions column_to_column;
 
    MultiplyDenseRowWithDenseColumnMatrixToRowOutputOptions column_to_row;
 
    MultiplyDenseRowWithDenseRowMatrixToColumnOutputOptions row_to_column;
 
    MultiplyDenseRowWithDenseRowMatrixToRowOutputOptions row_to_row;
};
 
inline void set_num_threads(MultiplyDenseRowWithDenseMatrixOptions& options, int num_threads) {
    options.column_to_column.num_threads = num_threads;
    options.column_to_row.num_threads = num_threads;
    options.row_to_column.num_threads = num_threads;
    options.row_to_row.num_threads = num_threads;
}
 
inline void set_dense_primary_block_size(MultiplyDenseRowWithDenseMatrixOptions& options, int primary_block_size) {
    options.column_to_column.primary_block_size = primary_block_size;
    options.column_to_row.primary_block_size = primary_block_size;
    options.row_to_column.primary_block_size = primary_block_size;
    options.row_to_row.primary_block_size = primary_block_size;
}
 
inline void set_dense_secondary_block_size(MultiplyDenseRowWithDenseMatrixOptions& options, int secondary_block_size) {
    options.column_to_column.secondary_block_size = secondary_block_size;
    options.column_to_row.secondary_block_size = secondary_block_size;
    options.row_to_column.secondary_block_size = secondary_block_size;
    options.row_to_row.secondary_block_size = secondary_block_size;
}
 
template<std::size_t accumulators_ = 4, typename LeftValue_, typename LeftIndex_, typename RightValue_, typename RightIndex_, typename Output_>
void multiply_dense_row_with_dense_matrix(
    const tatami::Matrix<LeftValue_, LeftIndex_>& left,
    const tatami::Matrix<RightValue_, RightIndex_>& right,
    Output_* const output,
    const bool output_row_major,
    const MultiplyDenseRowWithDenseMatrixOptions& options
) {
    if (right.prefer_rows()) {
        if (output_row_major) {
            multiply_dense_row_with_dense_row_matrix_to_row_output(left, right, output, options.row_to_row);
        } else {
            multiply_dense_row_with_dense_row_matrix_to_column_output(left, right, output, options.row_to_column);
        }
 
    } else {
        if (output_row_major) {
            multiply_dense_row_with_dense_column_matrix_to_row_output<accumulators_>(left, right, output, options.column_to_row);
        } else {
            multiply_dense_row_with_dense_column_matrix_to_column_output<accumulators_>(left, right, output, options.column_to_column);
        }
    }
}
 
}
 
#endif