Added missing set_device calls for nvidia_transform#52

Serge45 wants to merge 1 commit intoROCm:rocm_enabled_multi_backendROCm/bitsandbytes:rocm_enabled_multi_backendfrom

Serge45:feature/miss-set-device-transformSerge45/bitsandbytes:feature/miss-set-device-transformCopy head branch name to clipboard