reduce_sum.h

Go to the documentation of this file.

#pragma once

#include "tensor/tensor.h"
#include "cblas.h"
#include "utilities_internal.h"

#if defined(_HAS_CUDA_)
#include "magma.h"
#include "cudnn.h"

#include "cublas_v2.h"
#endif

namespace magmadnn {
namespace math {

template <typename T>
void reduce_sum(Tensor<T> *x, int axis, Tensor<T> *ones, Tensor<T> *out);

#if defined(_HAS_CUDA_)
struct reduce_sum_cudnn_settings_t {
    cudnnReduceTensorDescriptor_t descriptor;
    void *workspace;
    size_t workspace_size;
};

template <typename T>
void reduce_sum_device(Tensor<T> *x, int axis, Tensor<T> *out, reduce_sum_cudnn_settings_t settings);
#endif

}
}