correct printing of a IRCode that returns nothing

QuantumBFS · Jul 12, 2024 · cae04aa · cae04aa
1 parent 392c8dc
commit cae04aa
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/printing.jl b/src/printing.jl
@@ -63,3 +63,11 @@ function Base.show(io::IO, ::MIME"text/plain", g::Chain)
     print(io, ")")
 end
 
+function Base.show(io::IO, x::IRCode)
+  if x.stmts.stmt[1] isa Core.ReturnNode
+     printstyled(io, nothing)
+  else
+    printstyled(io, x)
+  end
+end
+