From c136efab658ea3ed7dfb7404d1a62c21cd72ec0a Mon Sep 17 00:00:00 2001
From: Oleksandr Pavlyk <21087696+oleksandr-pavlyk@users.noreply.github.com>
Date: Wed, 23 Jul 2025 15:32:27 -0500
Subject: [PATCH] Use absolute imports in cuda/nvbench/__init__.py

---
 python/cuda/nvbench/__init__.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/python/cuda/nvbench/__init__.py b/python/cuda/nvbench/__init__.py
index 3c320af..8091be2 100644
--- a/python/cuda/nvbench/__init__.py
+++ b/python/cuda/nvbench/__init__.py
@@ -18,22 +18,22 @@ except Exception as e:
 for libname in ("cupti", "nvperf_target", "nvperf_host"):
     load_nvidia_dynamic_lib(libname)
 
-from ._nvbench import (  # noqa: E402
+from cuda.nvbench._nvbench import (  # noqa: E402
     Benchmark as Benchmark,
 )
-from ._nvbench import (  # noqa: E402
+from cuda.nvbench._nvbench import (  # noqa: E402
     CudaStream as CudaStream,
 )
-from ._nvbench import (  # noqa: E402
+from cuda.nvbench._nvbench import (  # noqa: E402
     Launch as Launch,
 )
-from ._nvbench import (  # noqa: E402
+from cuda.nvbench._nvbench import (  # noqa: E402
     State as State,
 )
-from ._nvbench import (  # noqa: E402
+from cuda.nvbench._nvbench import (  # noqa: E402
     register as register,
 )
-from ._nvbench import (  # noqa: E402
+from cuda.nvbench._nvbench import (  # noqa: E402
     run_all_benchmarks as run_all_benchmarks,
 )