assert_egal mechanism

Author: Keno

base/compiler/ssair/ir.jl

@@ -272,7 +272,8 @@ function is_relevant_expr(e::Expr)
     return e.head in (:call, :invoke, :new, :(=), :(&),
                       :gc_preserve_begin, :gc_preserve_end,
                       :foreigncall, :isdefined, :copyast,
-                      :undefcheck, :throw_undef_if_not)
+                      :undefcheck, :throw_undef_if_not,
+                      :assert_egal)
 end
 
 function setindex!(x::UseRef, @nospecialize(v))

base/compiler/validation.jl

@@ -28,7 +28,8 @@ const VALID_EXPR_HEADS = IdDict{Any,Any}(
     :gc_preserve_begin => 0:typemax(Int),
     :gc_preserve_end => 0:typemax(Int),
     :thunk => 1:1,
-    :throw_undef_if_not => 2:2
+    :throw_undef_if_not => 2:2,
+    :assert_egal => 2:2
 )
 
 # @enum isn't defined yet, otherwise I'd use it for this

src/ast.c

@@ -67,7 +67,7 @@ jl_sym_t *macrocall_sym; jl_sym_t *colon_sym;
 jl_sym_t *hygienicscope_sym;
 jl_sym_t *escape_sym;
 jl_sym_t *gc_preserve_begin_sym; jl_sym_t *gc_preserve_end_sym;
-jl_sym_t *throw_undef_if_not_sym;
+jl_sym_t *throw_undef_if_not_sym; jl_sym_t *assert_egal_sym;
 
 static uint8_t flisp_system_image[] = {
 #include <julia_flisp.boot.inc>
@@ -392,6 +392,7 @@ void jl_init_frontend(void)
     generated_sym = jl_symbol("generated");
     generated_only_sym = jl_symbol("generated_only");
     throw_undef_if_not_sym = jl_symbol("throw_undef_if_not");
+    assert_egal_sym = jl_symbol("assert_egal");
 }
 
 JL_DLLEXPORT void jl_lisp_prompt(void)

src/codegen.cpp

@@ -3950,6 +3950,11 @@ static jl_cgval_t emit_expr(jl_codectx_t &ctx, jl_value_t *expr)
         undef_var_error_ifnot(ctx, ctx.builder.CreateTrunc(cond, T_int1), (jl_sym_t*)args[0]);
         return ghostValue(jl_void_type);
     }
+    else if (head == assert_egal_sym) {
+        Value *cond = emit_f_is(ctx, emit_expr(ctx, args[1]), emit_expr(ctx, args[2]));
+        undef_var_error_ifnot(ctx, ctx.builder.CreateTrunc(cond, T_int1), (jl_sym_t*)args[0]);
+        return ghostValue(jl_void_type);
+    }
     else if (head == invoke_sym) {
         return emit_invoke(ctx, ex);
     }

src/julia_internal.h

@@ -972,6 +972,7 @@ extern jl_sym_t *gc_preserve_begin_sym; extern jl_sym_t *gc_preserve_end_sym;
 extern jl_sym_t *generated_sym;
 extern jl_sym_t *generated_only_sym;
 extern jl_sym_t *throw_undef_if_not_sym;
+extern jl_sym_t *assert_egal_sym;
 
 struct _jl_sysimg_fptrs_t;