Skip to content

Commit 95c7ca7

Browse files
Nicoshevliligwu
authored andcommitted
Use kleidiAI on static builds (pytorch#3806)
Summary: X-link: https://github.com/facebookresearch/FBGEMM/pull/889 Pull Request resolved: pytorch#3806 Open-source builds as static library were failing Reviewed By: q10 Differential Revision: D71078076 fbshipit-source-id: d47321d2ea606d8d85a47f9c3697353dd9f325af
1 parent b7039eb commit 95c7ca7

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

include/fbgemm/FbgemmPackMatrixB.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -299,7 +299,7 @@ class PackedGemmMatrixB {
299299
bool pmat_passed_in{false};
300300
};
301301

302-
#ifndef FBGEMM_STATIC
302+
#ifndef _M_X64
303303

304304
template <>
305305
FBGEMM_API

src/FbgemmPackMatrixB.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ namespace fbgemm {
2828
// register layouts), 2 (with kernels of 1x2~6x2 register layout), or 3 (with
2929
// kernels of 1x3~4x3 register layout).
3030

31-
#ifndef FBGEMM_STATIC
31+
#ifndef _M_X64
3232

3333
template <>
3434
FBGEMM_API

0 commit comments

Comments
 (0)