updated gpu module API