1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
From b0b6611ae25441d27f4a0238d20bc3d77decaccc Mon Sep 17 00:00:00 2001
From: Hongxu Jia <hongxu.jia@windriver.com>
Date: Tue, 15 Oct 2024 14:25:30 +0800
Subject: [PATCH] disable avxvnni and avx512fp16 for x86
If host gcc is too old to support option -mavxvnni, the compile failed
...
DEBUG: | gcc: error: unrecognized command line option '-mavxvnni'; did you mean '-mavx512vnni'?
...
DEBUG: | gcc: error: unrecognized command-line option '-mavx512fp16'; did you mean '-mavx512bf16'?
...
Disable avxvnni and avx512fp16 for x86, use android as workaround
for native build
Upstream-Status: Inappropriate [Yocto specific]
Signed-off-by: Hongxu Jia <hongxu.jia@windriver.com>
---
tensorflow/workspace2.bzl | 1 +
.../0001-disable-avxvnni-and-avx512fp16.patch | 53 +++++++++++++++++++
2 files changed, 54 insertions(+)
create mode 100644 third_party/0001-disable-avxvnni-and-avx512fp16.patch
diff --git a/tensorflow/workspace2.bzl b/tensorflow/workspace2.bzl
index 43053967ca1..c48e09f2e38 100644
--- a/tensorflow/workspace2.bzl
+++ b/tensorflow/workspace2.bzl
@@ -152,6 +152,7 @@ def _tf_repositories():
name = "XNNPACK",
sha256 = "9e1efd0e43bff4a3b089faff6b0ee466291848fd855609a99e412befdec15bb0",
strip_prefix = "XNNPACK-bfea23551ac47c9d71b82120b299ff6173fc9097",
+ patch_file = ["//third_party:0001-disable-avxvnni-and-avx512fp16.patch"],
urls = tf_mirror_urls("https://github.com/google/XNNPACK/archive/bfea23551ac47c9d71b82120b299ff6173fc9097.zip"),
)
# LINT.ThenChange(//tensorflow/lite/tools/cmake/modules/xnnpack.cmake)
diff --git a/third_party/0001-disable-avxvnni-and-avx512fp16.patch b/third_party/0001-disable-avxvnni-and-avx512fp16.patch
new file mode 100644
index 00000000000..1c0d952070e
--- /dev/null
+++ b/third_party/0001-disable-avxvnni-and-avx512fp16.patch
@@ -0,0 +1,53 @@
+From 2e531e646757951cc799cf8a1f5f16409b1025ab Mon Sep 17 00:00:00 2001
+From: Hongxu Jia <hongxu.jia@windriver.com>
+Date: Tue, 15 Oct 2024 15:05:44 +0800
+Subject: [PATCH] disable avxvnni and avx512fp16
+
+Signed-off-by: Hongxu Jia <hongxu.jia@windriver.com>
+---
+ BUILD.bazel | 8 ++++----
+ 1 file changed, 4 insertions(+), 4 deletions(-)
+
+diff --git a/BUILD.bazel b/BUILD.bazel
+index b645c9d..f12debf 100644
+--- a/BUILD.bazel
++++ b/BUILD.bazel
+@@ -4477,7 +4477,7 @@ alias(
+ selects.config_setting_group(
+ name = "avx512vnnigfni_enabled_by_default",
+ match_any = [
+- "//build_config:x86",
++ "//build_config:android",
+ ],
+ )
+
+@@ -4493,7 +4493,7 @@ alias(
+ selects.config_setting_group(
+ name = "avx512amx_enabled_by_default",
+ match_any = [
+- "//build_config:x86",
++ "//build_config:android",
+ ],
+ )
+
+@@ -4510,7 +4510,7 @@ alias(
+ selects.config_setting_group(
+ name = "avx512fp16_enabled_by_default",
+ match_any = [
+- "//build_config:x86",
++ "//build_config:android",
+ ],
+ )
+
+@@ -4527,7 +4527,7 @@ alias(
+ selects.config_setting_group(
+ name = "avxvnni_enabled_by_default",
+ match_any = [
+- "//build_config:x86",
++ "//build_config:android",
+ ],
+ )
+
+--
+2.25.1
+
--
2.25.1
|
