Skip to content

Commit df36eb0

Browse files
penpornkcopybara-github
authored andcommitted
[xla:cpu] Fix build errors from ACL
Reported in: jax-ml/jax#26062 PiperOrigin-RevId: 719326299
1 parent 85743fe commit df36eb0

File tree

3 files changed

+37
-0
lines changed

3 files changed

+37
-0
lines changed
+23
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
diff --git a/src/BUILD.bazel b/src/BUILD.bazel
2+
index bf71e534e2..22377f1a32 100644
3+
--- a/src/BUILD.bazel
4+
+++ b/src/BUILD.bazel
5+
@@ -971,7 +971,6 @@ filegroup(
6+
"runtime/NEON/functions/NETranspose.cpp",
7+
"runtime/NEON/functions/NEUnstack.cpp",
8+
"runtime/NEON/functions/NEWinogradConvolutionLayer.cpp",
9+
- "runtime/OMP/OMPScheduler.cpp",
10+
"runtime/OffsetLifetimeManager.cpp",
11+
"runtime/OffsetMemoryPool.cpp",
12+
"runtime/OperatorTensor.cpp",
13+
@@ -984,6 +983,10 @@ filegroup(
14+
"runtime/Tensor.cpp",
15+
"runtime/TensorAllocator.cpp",
16+
"runtime/Utils.cpp"] +
17+
+ select({
18+
+ "//:openmp_flag": ["runtime/OMP/OMPScheduler.cpp"],
19+
+ "//conditions:default": [],
20+
+ }) +
21+
glob(["**/*.h",
22+
"**/*.hpp",
23+
"**/*.inl"]),

Diff for: third_party/compute_library/include_string.patch

+12
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,12 @@
1+
diff --git a/arm_compute/core/utils/logging/IPrinter.h b/arm_compute/core/utils/logging/IPrinter.h
2+
index 42dca58ea1..9c8b1598e8 100644
3+
--- a/arm_compute/core/utils/logging/IPrinter.h
4+
+++ b/arm_compute/core/utils/logging/IPrinter.h
5+
@@ -25,6 +25,7 @@
6+
#define ARM_COMPUTE_LOGGING_PRINTER_H
7+
8+
#include "support/Mutex.h"
9+
+#include <string>
10+
11+
namespace arm_compute
12+
{

Diff for: workspace2.bzl

+2
Original file line numberDiff line numberDiff line change
@@ -179,6 +179,8 @@ def _tf_repositories():
179179
patch_file = [
180180
"//third_party/compute_library:compute_library.patch",
181181
"//third_party/compute_library:acl_thread_local_scheduler.patch",
182+
"//third_party/compute_library:exclude_omp_scheduler.patch",
183+
"//third_party/compute_library:include_string.patch",
182184
],
183185
sha256 = "c4ca329a78da380163b2d86e91ba728349b6f0ee97d66e260a694ef37f0b0d93",
184186
strip_prefix = "ComputeLibrary-23.05.1",

0 commit comments

Comments
 (0)