Does AIT handle if/else in forward function?

[jiangwei221]

Hi AIT team:

Sorry to bring up this stupid question, but does AIT handles if/else in the forward function?
For example

if x2.mean() > 0:
    x = x + x2
else:
    x = x - x2

If I supply x2 with positive mean when compiling, will the compiled model work with both branches?

Thanks!

[ColinPeppler]

Hi Jiang, is x2 produced by an operator on the graph? If yes, then that's data-dependent control flow and AIT doesn't support that.

One workaround I can think of is this:

x = x + x2 * sign(x2.mean())