Merge pull request #7993 from kmaehashi/fix-7967

Add lazy load for `cupyx.lapack`
cupy · Nov 28, 2023 · 1a04141 · 1a04141
2 parents 4f0687e + 8aaf434
commit 1a04141
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/cupyx/__init__.py b/cupyx/__init__.py
@@ -22,3 +22,12 @@
 from cupyx._pinned_array import zeros_like_pinned  # NOQA
 
 from cupyx._gufunc import GeneralizedUFunc  # NOQA
+
+
+def __getattr__(key):
+    if key == 'lapack':
+        import cupyx.lapack
+        return cupyx.lapack
+
+    raise AttributeError(
+        "module '{}' has no attribute '{}'".format(__name__, key))