Add gpu bench and remove unneeded code