cancel, defer, errdefer all working as intended now

1 files changed, 8 insertions, 0 deletions

diff --git a/src/ir_print.cpp b/src/ir_print.cpp
index 0348cfc986..8b8445f625 100644
--- a/src/ir_print.cpp
+++ b/src/ir_print.cpp
@@ -1550,6 +1550,11 @@ static void ir_print_await_gen(IrPrint *irp, IrInstructionAwaitGen *instruction)
     fprintf(irp->f, ")");
 }
 
+static void ir_print_test_cancel_requested(IrPrint *irp, IrInstructionTestCancelRequested *instruction) {
+    const char *arg = instruction->use_return_begin_prev_value ? "UseReturnBeginPrevValue" : "AdditionalCheck";
+    fprintf(irp->f, "@testCancelRequested(%s)", arg);
+}
+
 static void ir_print_instruction(IrPrint *irp, IrInstruction *instruction) {
     ir_print_prefix(irp, instruction);
     switch (instruction->id) {
@@ -2032,6 +2037,9 @@ static void ir_print_instruction(IrPrint *irp, IrInstruction *instruction) {
         case IrInstructionIdAwaitGen:
             ir_print_await_gen(irp, (IrInstructionAwaitGen *)instruction);
             break;
+        case IrInstructionIdTestCancelRequested:
+            ir_print_test_cancel_requested(irp, (IrInstructionTestCancelRequested *)instruction);
+            break;
     }
     fprintf(irp->f, "\n");
 }