Move GPU traits into separate headers