Revert "Reverting the PR adding Kleidiai-based int4 kernels (#145392)" (#145505)

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 0940eb6d44f3cf69dd840db990245cbe1f78e770 (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
This commit is contained in:
Nikhil Gupta
2025-01-23 18:50:58 +00:00
committed by PyTorch MergeBot
parent 34b8d8b0c0
commit 41b38f755c
37 changed files with 1940 additions and 23 deletions

View File

@ -257,6 +257,7 @@ filegroup(
# target that generates these sources...
)
# TODO: Enable support for KleidiAI bazel build
header_template_rule(
name = "aten_src_ATen_config",
src = "aten/src/ATen/Config.h.in",
@ -276,6 +277,7 @@ header_template_rule(
"@AT_PARALLEL_NATIVE@": "1",
"@AT_BLAS_F2C@": "0",
"@AT_BLAS_USE_CBLAS_DOT@": "1",
"@AT_KLEIDIAI_ENABLED@": "0",
},
)