Merge pull request #25 from gomlx/value-opname

Added Value.OpName() to allow some introspection into the computation

Author: janpfeifer

docs/CHANGELOG.md

@@ -5,7 +5,8 @@
 
 ## StableHLO
 
-- Added `Call()` op (thx @ajroetker)
+- Added `Call()` op (thx @ajroetker).
+- Added `Value.OpName()` for introspection.
 - Dynamic Shapes (thx @ajroetker) -- with and without dynamic bounds.
   - Added `DimensionBounds` and `EncodeBounds` fields to `Shape` struct
   - Added ops: `DynamicReshape`, `DynamicBroadcastInDim`, `DynamicIota`, `DynamicGather`,

pkg/stablehlo/value.go

@@ -49,6 +49,15 @@ func (v *Value) String() string {
 	return "%" + v.name
 }
 
+// OpName returns the name of the operation that created this value.
+func (v *Value) OpName() string {
+	if v.stmt == nil {
+		return "nil"
+	}
+	return v.stmt.OpType.String()
+}
+
+
 // ConvertToValidName replaces any characters not in { "0"-"9", "a"-"z", "A-Z", "_" } to a "_",
 // making it a valid name for values and function arguments.
 func ConvertToValidName(name string) string {