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

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

Browse Source 
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
Download Patch File Download Diff File Expand all files Collapse all files

4
buckbuild.bzl
View File

@ -1074,6 +1074,7 @@ def define_buck_targets(
],
)
# TODO: Enable support for KleidiAI bazel build
# @lint-ignore BUCKLINT
fb_native.genrule(
name = "generate_aten_config",
@ -1126,6 +1127,9 @@ def define_buck_targets(
"--replace",
"@AT_BLAS_USE_CBLAS_DOT@",
"AT_BLAS_USE_CBLAS_DOT_FBXPLAT",
"--replace",
"@AT_KLEIDIAI_ENABLED@",
"0",
]),
outs = {
"Config.h": ["Config.h"],