jorisvandenbossche commented on code in PR #40385: URL: https://github.com/apache/arrow/pull/40385#discussion_r1541309381
########## python/pyarrow/__init__.py: ########## @@ -281,6 +281,13 @@ def print_entry(label, value): import pyarrow.types as types +try: + # Try importing the cuda module to ensure libarrow_cuda gets loaded + # to register the CUDA device for the C Data Interface import + import pyarrow.cuda +except ImportError: Review Comment: > I would be more worried by the potential implications of bringing CUDA into the process, though that's totally unsubstantiated. For example worry about the memory usage increase (in case it wasn't actually needed)? > Ok, so let's do it on demand if required? You mean actually first check the device type in the struct? -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
