File tree Expand file tree Collapse file tree 1 file changed +4
-3
lines changed
src/lightning/fabric/utilities Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change 12
12
# See the License for the specific language governing permissions and
13
13
# limitations under the License.
14
14
"""General utilities."""
15
+ import functools
15
16
import operator
16
17
import platform
17
18
import sys
36
37
37
38
38
39
@functools .lru_cache (maxsize = 1 )
39
- def _lightning_habana_available () -> bool :
40
- # This is defined as a function instead of a constant to avoid circular imports, because `lightning_habana `
40
+ def _lightning_xpu_available () -> bool :
41
+ # This is defined as a function instead of a constant to avoid circular imports, because `lightning_xpu `
41
42
# also imports Lightning
42
- return bool (RequirementCache ("lightning-habana " )) and _try_import_module ("lightning_habana " )
43
+ return bool (RequirementCache ("lightning-xpu " )) and _try_import_module ("lightning_xpu " )
You can’t perform that action at this time.
0 commit comments