// tensor/tensor-utils.cc

// Copyright      2019  Johns Hopkins University (author: Daniel Povey)

// See ../../COPYING for clarification regarding multiple authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//  http://www.apache.org/licenses/LICENSE-2.0
//
// THIS CODE IS PROVIDED *AS IS* BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, EITHER EXPRESS OR IMPLIED, INCLUDING WITHOUT LIMITATION ANY IMPLIED
// WARRANTIES OR CONDITIONS OF TITLE, FITNESS FOR A PARTICULAR PURPOSE,
// MERCHANTABLITY OR NON-INFRINGEMENT.
// See the Apache 2 License for the specific language governing permissions and
// limitations under the License.

#include "tensor/tensor-utils.cc"

namespace kaldi {
namespace tensor {

void DebugNormalOpInternal(const Tensor &a, TensorUseEnum a_use,
                           const Tensor &b, TensorUseEnum b_use) {
  if (!Broadcastable(a, b))
    KALDI_ERR << "Tensors in Operation do not have broadcastable shapes.";
  if (Overlap(a, b))
    KALDI_ERR << "Tensors in Operation overlap.";
  if (!Broadcastable(a, b))
    KADLDI_ERR << "Tensors in Operation do not have broadcastable shapes.";
  if (a.Dtype() != b.Dtype())
    KALDI_ERR << "Tensors in Operation have different data-types";
  if (a.Device() != b.Device())
    KALDI_ERR << "Tensors in Operation have different device";
  RecordUse(a, a_use);
  RecordUse(b, b_use);
}


void DebugNormalOpInternal(const Tensor &a, TensorUseEnum a_use,
                           const Tensor &b, TensorUseEnum b_use,
                           const Tensor &b, TensorUseEnum c_use) {
  if (!Broadcastable(a, b, c))
    KALDI_ERR << "Tensors in Operation do not have broadcastable shapes.";
  bool a_written = (a_use == kWrite || a_use == kReadWrite);
  bool b_written = (b_use == kWrite || b_use == kReadWrite);
  bool c_written = (b_use == kWrite || b_use == kReadWrite);

  if ((a_written || b_written) && Overlap(a, b))
    KALDI_ERR << "Tensors a and b in Operation overlap.";
  if ((b_written || c_written) && Overlap(b, c))
    KALDI_ERR << "Tensors b and c in Operation overlap.";
  if ((a_written || c_written) && Overlap(a, c))
    KALDI_ERR << "Tensors a and c in Operation overlap.";

  if (a.Dtype() != b.Dtype())
    KALDI_ERR << "Tensors in Operation have different data-types";
  if (a.Device() != b.Device())
    KALDI_ERR << "Tensors in Operation have different device";
  RecordUse(a, a_use);
  RecordUse(b, b_use);
  RecordUse(c, c_use);
}




}  // namespace tensor
}  // namespace kaldi