Merge "Upgrade pthreadpool to 344531b40881b1ee41508a9c70c8fbbef3bd6cad" am: 648365d11d am: 7b7f865f8e

Original change: https://android-review.googlesource.com/c/platform/external/pthreadpool/+/1520473

MUST ONLY BE SUBMITTED BY AUTOMERGER

Change-Id: I58f93853b625f5ebe5b111c246d42ca1e3f2cc83

1 files changed, 55 insertions, 0 deletions

diff --git a/src/shim.c b/src/shim.c
index e90ac45..39ec884 100644
--- a/src/shim.c
+++ b/src/shim.c
@@ -383,6 +383,61 @@ void pthreadpool_parallelize_5d_tile_2d(
 	}
 }
 
+void pthreadpool_parallelize_6d(
+	pthreadpool_t threadpool,
+	pthreadpool_task_6d_t task,
+	void* argument,
+	size_t range_i,
+	size_t range_j,
+	size_t range_k,
+	size_t range_l,
+	size_t range_m,
+	size_t range_n,
+	uint32_t flags)
+{
+	for (size_t i = 0; i < range_i; i++) {
+		for (size_t j = 0; j < range_j; j++) {
+			for (size_t k = 0; k < range_k; k++) {
+				for (size_t l = 0; l < range_l; l++) {
+					for (size_t m = 0; m < range_m; m++) {
+						for (size_t n = 0; n < range_n; n++) {
+							task(argument, i, j, k, l, m, n);
+						}
+					}
+				}
+			}
+		}
+	}
+}
+
+void pthreadpool_parallelize_6d_tile_1d(
+	pthreadpool_t threadpool,
+	pthreadpool_task_6d_tile_1d_t task,
+	void* argument,
+	size_t range_i,
+	size_t range_j,
+	size_t range_k,
+	size_t range_l,
+	size_t range_m,
+	size_t range_n,
+	size_t tile_n,
+	uint32_t flags)
+{
+	for (size_t i = 0; i < range_i; i++) {
+		for (size_t j = 0; j < range_j; j++) {
+			for (size_t k = 0; k < range_k; k++) {
+				for (size_t l = 0; l < range_l; l++) {
+					for (size_t m = 0; m < range_m; m++) {
+						for (size_t n = 0; n < range_n; n += tile_n) {
+							task(argument, i, j, k, l, m, n, min(range_n - n, tile_n));
+						}
+					}
+				}
+			}
+		}
+	}
+}
+
 void pthreadpool_parallelize_6d_tile_2d(
 	pthreadpool_t threadpool,
 	pthreadpool_task_6d_tile_2d_t task,