Set newly created context as current

This commit is contained in:
Andrzej Janik
2025-07-24 18:04:45 +00:00
parent 5deada8426
commit 893bbe80d9

View File

@ -188,6 +188,8 @@ pub(crate) unsafe fn create_v2(
dev: cuda_types::cuda::CUdevice, dev: cuda_types::cuda::CUdevice,
) -> CUresult { ) -> CUresult {
let handle = Context::wrap(Context::new(dev)); let handle = Context::wrap(Context::new(dev));
// TODO: optimize
set_current(handle)?;
*ctx = handle; *ctx = handle;
Ok(()) Ok(())
} }