mean_squared_error.hpp

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2019-2025, NVIDIA CORPORATION.
 * SPDX-License-Identifier: Apache-2.0
 */
#pragma once
 
#include "../matrix/data.hpp"
#include "../matrix/part_descriptor.hpp"
 
#include <raft/core/comms.hpp>
#include <raft/core/device_mdspan.hpp>
 
namespace MLCommon {
namespace LinAlg {
namespace opg {
 
void meanSquaredError(double* out,
                      const Matrix::Data<double>& in1,
                      const Matrix::PartDescriptor& in1Desc,
                      const Matrix::Data<double>& in2,
                      const Matrix::PartDescriptor& in2Desc,
                      const raft::comms::comms_t& comm,
                      cudaStream_t stream,
                      int root             = 0,
                      bool broadcastResult = true);
void meanSquaredError(float* out,
                      const Matrix::Data<float>& in1,
                      const Matrix::PartDescriptor& in1Desc,
                      const Matrix::Data<float>& in2,
                      const Matrix::PartDescriptor& in2Desc,
                      const raft::comms::comms_t& comm,
                      cudaStream_t stream,
                      int root             = 0,
                      bool broadcastResult = true);
 
}  // end namespace opg
}  // end namespace LinAlg
}  // end namespace MLCommon