Skip to content

Commit

Permalink
Make sure librmm is imported by rmm
Browse files Browse the repository at this point in the history
  • Loading branch information
vyasr committed Feb 2, 2025
1 parent 20b1994 commit 44306de
Showing 1 changed file with 8 additions and 1 deletion.
9 changes: 8 additions & 1 deletion python/rmm/rmm/__init__.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright (c) 2018-2024, NVIDIA CORPORATION.
# Copyright (c) 2018-2025, NVIDIA CORPORATION.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
Expand All @@ -14,6 +14,13 @@

import warnings

try:
import librmm
except ModuleNotFoundError:
pass
else:
del librmm

from rmm import mr
from rmm._version import __git_commit__, __version__
from rmm.mr import disable_logging, enable_logging, get_log_filenames
Expand Down

0 comments on commit 44306de

Please sign in to comment.