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 }