Add gpu prod implementation.
code clean up.
remove import not used.