1 /*
   2  * Copyright (c) 2019, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.
   8  *
   9  * This code is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  12  * version 2 for more details (a copy is included in the LICENSE file that
  13  * accompanied this code).
  14  *
  15  * You should have received a copy of the GNU General Public License version
  16  * 2 along with this work; if not, write to the Free Software Foundation,
  17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  18  *
  19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  20  * or visit www.oracle.com if you need additional information or have any
  21  * questions.
  22  */
  23 
  24 
  25 package org.graalvm.compiler.replacements.nodes;
  26 
  27 import static org.graalvm.compiler.nodeinfo.NodeCycles.CYCLES_2;
  28 import static org.graalvm.compiler.nodeinfo.NodeSize.SIZE_1;
  29 
  30 import org.graalvm.compiler.core.common.type.FloatStamp;
  31 import org.graalvm.compiler.core.common.type.Stamp;
  32 import org.graalvm.compiler.graph.NodeClass;
  33 import org.graalvm.compiler.graph.spi.CanonicalizerTool;
  34 import org.graalvm.compiler.lir.gen.ArithmeticLIRGeneratorTool;
  35 import org.graalvm.compiler.nodeinfo.NodeInfo;
  36 import org.graalvm.compiler.nodes.ConstantNode;
  37 import org.graalvm.compiler.nodes.NodeView;
  38 import org.graalvm.compiler.nodes.ValueNode;
  39 import org.graalvm.compiler.nodes.calc.TernaryNode;
  40 import org.graalvm.compiler.nodes.spi.ArithmeticLIRLowerable;
  41 import org.graalvm.compiler.nodes.spi.NodeLIRBuilderTool;
  42 
  43 import jdk.vm.ci.meta.JavaConstant;
  44 import jdk.vm.ci.meta.JavaKind;
  45 import org.graalvm.compiler.serviceprovider.GraalServices;
  46 
  47 @NodeInfo(cycles = CYCLES_2, size = SIZE_1)
  48 public final class FusedMultiplyAddNode extends TernaryNode implements ArithmeticLIRLowerable {
  49 
  50     public static final NodeClass<FusedMultiplyAddNode> TYPE = NodeClass.create(FusedMultiplyAddNode.class);
  51 
  52     public FusedMultiplyAddNode(ValueNode a, ValueNode b, ValueNode c) {
  53         super(TYPE, computeStamp(a.stamp(NodeView.DEFAULT), b.stamp(NodeView.DEFAULT), c.stamp(NodeView.DEFAULT)), a, b, c);
  54         assert a.getStackKind().isNumericFloat();
  55         assert b.getStackKind().isNumericFloat();
  56         assert c.getStackKind().isNumericFloat();
  57     }
  58 
  59     @Override
  60     public Stamp foldStamp(Stamp stampX, Stamp stampY, Stamp stampZ) {
  61         return computeStamp(stampX, stampY, stampZ);
  62     }
  63 
  64     private static Stamp computeStamp(Stamp stampX, Stamp stampY, Stamp stampZ) {
  65         Stamp m = FloatStamp.OPS.getMul().foldStamp(stampX, stampY);
  66         return FloatStamp.OPS.getAdd().foldStamp(m, stampZ);
  67     }
  68 
  69     @Override
  70     public ValueNode canonical(CanonicalizerTool tool, ValueNode a, ValueNode b, ValueNode c) {
  71         if (a.isConstant() && b.isConstant() && c.isConstant()) {
  72             JavaConstant ca = a.asJavaConstant();
  73             JavaConstant cb = b.asJavaConstant();
  74             JavaConstant cc = c.asJavaConstant();
  75 
  76             ValueNode res;
  77             if (a.getStackKind() == JavaKind.Float) {
  78                 res = ConstantNode.forFloat(GraalServices.fma(ca.asFloat(), cb.asFloat(), cc.asFloat()));
  79             } else {
  80                 assert a.getStackKind() == JavaKind.Double;
  81                 res = ConstantNode.forDouble(GraalServices.fma(ca.asDouble(), cb.asDouble(), cc.asDouble()));
  82             }
  83             return res;
  84         }
  85         return this;
  86     }
  87 
  88     @Override
  89     public void generate(NodeLIRBuilderTool builder, ArithmeticLIRGeneratorTool gen) {
  90         builder.setResult(this, gen.emitFusedMultiplyAdd(builder.operand(getX()), builder.operand(getY()), builder.operand(getZ())));
  91     }
  92 }