mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
41b38f755c
https://github.com/pytorch/pytorch/pull/134124 was reverted by https://github.com/pytorch/pytorch/pull/145392 due to KleidiAI clone issue.
1. This reverts commit 0940eb6d44 (https://github.com/pytorch/pytorch/pull/145392 )and Fixes KleidiAI mirror issue.
2. KleidiAI is now cloned from github mirror instead of arm gitlab
Change-Id: I7d6eee7214cd117d3057d615936fcc3ee6052fa2
Fixes https://github.com/pytorch/pytorch/issues/145273
Pull Request resolved: https://github.com/pytorch/pytorch/pull/145505
Approved by: https://github.com/malfet
8 lines
162 B
Python
8 lines
162 B
Python
# mypy: allow-untyped-defs
|
|
import torch
|
|
|
|
|
|
def is_available():
|
|
r"""Return whether PyTorch is built with KleidiAI support."""
|
|
return torch._C._has_kleidiai
|