Set newly created context as current (#431)

This commit is contained in:
Andrzej Janik
2025-07-24 20:05:04 +02:00
committed by GitHub
parent 5deada8426
commit e8e20294a6

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(())
} }