Update gpumem.go