Skip to content

Commit b75ebbd

Browse files
committed
[aarch64] add mkldnn fix for torch.compile()
1 parent b92da8c commit b75ebbd

File tree

1 file changed

+24
-0
lines changed

1 file changed

+24
-0
lines changed
Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
cpu: aarch64: remove weight format checking
2+
3+
with weights pre-packing enabled in torch.compile(),
4+
the weights come already reorderd and in oneDNN format,
5+
so, checking for format::any is not valid.
6+
---
7+
src/cpu/aarch64/acl_inner_product.hpp | 1 -
8+
1 file changed, 1 deletion(-)
9+
10+
diff --git a/src/cpu/aarch64/acl_inner_product.hpp b/src/cpu/aarch64/acl_inner_product.hpp
11+
index a2be164f0..eadeb92b6 100644
12+
--- a/src/cpu/aarch64/acl_inner_product.hpp
13+
+++ b/src/cpu/aarch64/acl_inner_product.hpp
14+
@@ -101,7 +101,6 @@ struct acl_inner_product_fwd_t : public primitive_t {
15+
primitive_attr_t::skip_mask_t::post_ops, f32);
16+
const bool ok = is_fwd() && !has_zero_dim_memory()
17+
&& utils::one_of(true, is_fp16_ok, is_fp32_ok)
18+
- && weights_md_.format_kind == format_kind::any
19+
&& set_default_params() == status::success;
20+
21+
if (!ok) return status::unimplemented;
22+
--
23+
2.34.1
24+

0 commit comments

Comments
 (0)