[math] enable debugging for new style of transformations in BrainPy

[chaoming0625]

The previous version of brainpy >=2.4.0, <=2.4.1 has a redundant (or event wrong) compilation phases. For example, for the following code:

import brainpy.math as bm


@bm.jit
def f2(a):
  print('f2 ...')
  return a


@bm.jit
def f1(a):
  print('f1 ...')
  return f2(a)


f1(1.)

In brainpy==2.4.1 you will got:

f1 ...
f2 ...
f2 ...
f1 ...

It first evaluates f1, then evaluate f2, then compile f2, finally compiling f1. This is not the correct order.

For this version of brainpy, we got:

f1 ...
f2 ...
f1 ...
f2 ...

It first evaluates f1, then evaluate f2, then compile f1, finally compilef2. This is exactly what we want!