fix cuda stuff