// Copyright (c) 2019 Graphcore Ltd. All rights reserved.
#include <string>
#include <vector>
#include <popart/op/abs.hpp>
#include <popart/op/cos.hpp>
#include <popart/op/sign.hpp>
#include <popart/op/sin.hpp>
#include <popart/patterns/elementwisegradoppattern.hpp>
#include <popart/patterns/patterns.hpp>

namespace popart {

namespace {
// Replace a AbsGradOp with
// (fwd_in) -> [Sign] -> (tmp1)
// {(tmp1), (grad_in)} -> [Mul] -> (grad_out)
static PatternCreator<ElementWiseGradOpPattern<AbsGradOp, SignOp>>
    sinGradOpPattern("AbsGradOp",
                     /* enabled = */ true,
                     /* mandatory = */ true);

// Replace a SinGradOp with
// (fwd_in) -> [Cos] -> (tmp1)
// {(tmp1), (grad_in)} -> [Mul] -> (grad_out)
static PatternCreator<ElementWiseGradOpPattern<SinGradOp, CosOp>>
    absGradOpPattern("SinGradOp",
                     /* enabled = */ true,
                     /* mandatory = */ true);

} // namespace

} // namespace popart