1. Make while_op accept GPU conditional data
2. Add more complex test cases for while_loop API

Add basic while_loop