Initial GPU version