diff --git a/cuda_core/cuda/core/experimental/_event.py b/cuda_core/cuda/core/experimental/_event.py index 2551365aa..382384a65 100644 --- a/cuda_core/cuda/core/experimental/_event.py +++ b/cuda_core/cuda/core/experimental/_event.py @@ -8,7 +8,7 @@ from dataclasses import dataclass from typing import TYPE_CHECKING, Optional -from cuda.core.experimental._utils.cuda_utils import check_or_create_options, driver, handle_return +from cuda.core.experimental._utils.cuda_utils import CUDAError, check_or_create_options, driver, handle_return if TYPE_CHECKING: import cuda.bindings