Skip to content

Commit 8706d3f

Browse files
authored
Fix internal test_linear_8bit_act_xbit_weightAppleMac
Differential Revision: D70186827 Pull Request resolved: #1776
1 parent 98c4e2e commit 8706d3f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

torchao/experimental/ops/tests/test_linear_8bit_act_xbit_weight.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,13 +14,13 @@
1414

1515
#if defined(TORCHAO_ENABLE_KLEIDI)
1616
#include <torchao/experimental/kernels/cpu/aarch64/kleidi/kai_matmul_clamp_f32_qai8dxp_qsi4c32p.h>
17+
using namespace torchao::kernels::cpu::aarch64::kleidi::
18+
kai_matmul_clamp_f32_qai8dxp_qsi4c32p;
1719
#endif // TORCHAO_ENABLE_KLEIDI
1820

1921
const float kTol = 1.0e-5;
2022

2123
using namespace torchao::ops::linear_8bit_act_xbit_weight;
22-
using namespace torchao::kernels::cpu::aarch64::kleidi::
23-
kai_matmul_clamp_f32_qai8dxp_qsi4c32p;
2424

2525
template <int weight_nbit, bool has_weight_zeros, bool has_bias, bool has_clamp>
2626
UKernelConfig get_ukernel_config() {

0 commit comments

Comments
 (0)