Getting Numba compiler internal structures

numba_internals.py

from numba import bytecode, interpreter, ir

def foo(a, b):
    c = 0
    for i in range(a):
        for j in range(b):
            c += i + j
    return c


bc = bytecode.ByteCode(func=foo)
intrp =interpreter.Interpreter(bytecode=bc)
intrp.interpret()

print("CFG info".center(80, '='))
intrp.cfa.dump()  # CFG

print("DFG info".center(80, '='))
intrp.dfa.dump()

intrp.dump() # Dump the Numba IR

print("Getting the entry block".center(80, '='))
firstblock = intrp.blocks[intrp.cfa.blockseq[0]]  # entry block

firstblock.dump()