[small BUG] We where loosing the condition when moving Assert op to the GPU
Fix test caused by new optimization.
Fix test in FAST_COMPILE.