1 /*
  2  * Copyright (c) 2025, 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.  Oracle designates this
  8  * particular file as subject to the "Classpath" exception as provided
  9  * by Oracle in the LICENSE file that accompanied this code.
 10  *
 11  * This code is distributed in the hope that it will be useful, but WITHOUT
 12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 14  * version 2 for more details (a copy is included in the LICENSE file that
 15  * accompanied this code).
 16  *
 17  * You should have received a copy of the GNU General Public License version
 18  * 2 along with this work; if not, write to the Free Software Foundation,
 19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 20  *
 21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
 22  * or visit www.oracle.com if you need additional information or have any
 23  * questions.
 24  */
 25 package hat.dialect;
 26 
 27 import jdk.incubator.code.CodeContext;
 28 import jdk.incubator.code.TypeElement;
 29 import jdk.incubator.code.Value;
 30 
 31 import java.util.List;
 32 import java.util.Map;
 33 
 34 public abstract class HATF16BinaryOp extends HATF16Op {
 35 
 36     protected final TypeElement elementType;
 37     protected final BinaryOpType operationType;
 38     protected final List<Boolean> references;
 39     private final ReducedFloatType reducedFloatType;
 40     protected final byte f32;
 41 
 42     public static final byte FIRST_OP = 0x01;
 43     public static final byte LAST_OP = 0x10;
 44 
 45     public enum BinaryOpType {
 46         ADD("+"),
 47         SUB("-"),
 48         MUL("*"),
 49         DIV("/");
 50 
 51         String symbol;
 52 
 53         BinaryOpType(String symbol) {
 54             this.symbol = symbol;
 55         }
 56 
 57         public String symbol() {
 58             return symbol;
 59         }
 60     }
 61 
 62     public HATF16BinaryOp(TypeElement typeElement, ReducedFloatType reducedFloatType, BinaryOpType operationType, List<Boolean> references, byte f32, List<Value> operands) {
 63         super("", operands);
 64         this.elementType = typeElement;
 65         this.operationType = operationType;
 66         this.references = references;
 67         this.f32 = f32;
 68         this.reducedFloatType = reducedFloatType;
 69     }
 70 
 71     public HATF16BinaryOp(HATF16BinaryOp op, CodeContext copyContext) {
 72         super(op, copyContext);
 73         this.elementType = op.elementType;
 74         this.operationType = op.operationType;
 75         this.references = op.references;
 76         this.f32 = op.f32;
 77         this.reducedFloatType = op.reducedFloatType;
 78     }
 79 
 80     @Override
 81     public TypeElement resultType() {
 82         return this.elementType;
 83     }
 84 
 85     @Override
 86     public Map<String, Object> externalize() {
 87         return Map.of("hat.dialect.fp16." + varName(), operationType.symbol());
 88     }
 89 
 90     public BinaryOpType binaryOperationType() {
 91         return operationType;
 92     }
 93 
 94     public List<Boolean> references() {
 95         return references;
 96     }
 97 
 98     public byte getF32() {
 99         return f32;
100     }
101 
102     public ReducedFloatType reducedFloatType() {
103         return reducedFloatType;
104     }
105 }