GPU compilation fix