Fix malloc size for device sse.