frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add torch.backends.mkldnn.is_acl_available() method (#165678)

Browse Source 
That tells whether or not PyTorch was compiled with Arm Compute Library
Pull Request resolved: https://github.com/pytorch/pytorch/pull/165678
Approved by: https://github.com/Skylion007, https://github.com/atalman, https://github.com/albanD
ghstack dependencies: #165583, #165584, #165676
This commit is contained in:
Nikita Shulga
2025-10-16 13:14:17 -07:00
committed by PyTorch MergeBot
parent 4d833f859b
commit ce109b3f79
3 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
torch/_C/__init__.pyi.in
View File

@ -1442,6 +1442,7 @@ _has_cuda: _bool
_has_magma: _bool _has_magma: _bool
_has_xpu: _bool _has_xpu: _bool
_has_mkldnn: _bool _has_mkldnn: _bool
_has_mkldnn_acl: _bool
_has_cudnn: _bool _has_cudnn: _bool
_has_cusparselt: _bool _has_cusparselt: _bool
has_spectral: _bool has_spectral: _bool

5
torch/backends/mkldnn/__init__.py
View File

@ -19,6 +19,11 @@ def is_available():
return torch._C._has_mkldnn return torch._C._has_mkldnn
def is_acl_available():
r"""Return whether PyTorch is built with MKL-DNN + ACL support."""
return torch._C._has_mkldnn_acl
VERBOSE_OFF = 0 VERBOSE_OFF = 0
VERBOSE_ON = 1 VERBOSE_ON = 1
VERBOSE_ON_CREATION = 2 VERBOSE_ON_CREATION = 2

2
torch/csrc/Module.cpp
View File

@ -2701,6 +2701,8 @@ Call this whenever a new thread is created in order to propagate values from
ASSERT_TRUE(set_module_attr("_has_xpu", has_xpu)); ASSERT_TRUE(set_module_attr("_has_xpu", has_xpu));
ASSERT_TRUE( ASSERT_TRUE(
set_module_attr("_has_mkldnn", at::hasMKLDNN() ? Py_True : Py_False)); set_module_attr("_has_mkldnn", at::hasMKLDNN() ? Py_True : Py_False));
ASSERT_TRUE(set_module_attr(
"_has_mkldnn_acl", AT_MKLDNN_ACL_ENABLED() ? Py_True : Py_False));
ASSERT_TRUE(set_module_attr("_GLIBCXX_USE_CXX11_ABI", Py_True)); ASSERT_TRUE(set_module_attr("_GLIBCXX_USE_CXX11_ABI", Py_True));