Add dump function to jit module