feat: more constant checks

Author: avik-pal

src/Ops.jl

@@ -206,6 +206,12 @@ for (T, mlir_func) in (
 
             splatattr = MLIR.API.$mlir_func(tt, number)
             cst_op = stablehlo.constant(; output=tt, value=splatattr, location=location)
+
+            parent_func_op = MLIR.IR.get_parent_of_type_function_op(cst_op)
+            if parent_func_op == C_NULL
+                error("Constant must be created inside a Function Op.")
+            end
+
             cst = MLIR.IR.result(cst_op)
             ta = TracedRArray{$T,length(shape)}((), cst, shape)
             return ta
@@ -226,6 +232,12 @@ end
     tt = MLIR.IR.TensorType(shape, MLIR.IR.Type(T))
     splatattr = MLIR.API.mlirDenseElementsAttrSplatGet(tt, _fill_element_attr(element))
     cst_op = stablehlo.constant(; output=tt, value=splatattr, location=location)
+
+    parent_func_op = MLIR.IR.get_parent_of_type_function_op(cst_op)
+    if parent_func_op == C_NULL
+        error("Constant must be created inside a Function Op.")
+    end
+
     cst = MLIR.IR.result(cst_op)
     ta = TracedRArray{T,length(shape)}((), cst, shape)
     return ta