* add flag

* change flag

* use prim flag

* fix code

* fix softmax prim flag

* set case timeout