From: Alyssa Rosenzweig <alyssa.rosenzweig@collabora.com>
Date: Mon, 9 Dec 2019 14:00:24 +0000 (-0500)
Subject: panfrost: Add routines to calculate stack size/shift
X-Git-Url: https://git.libre-soc.org/?a=commitdiff_plain;ds=sidebyside;h=eac9247b2dfebdb22a3446f588b00ec773b9e3fa;p=mesa.git

panfrost: Add routines to calculate stack size/shift

These implement the aforementioned formulas.

Signed-off-by: Alyssa Rosenzweig <alyssa.rosenzweig@collabora.com>
---

diff --git a/src/panfrost/encoder/pan_encoder.h b/src/panfrost/encoder/pan_encoder.h
index ceff2e949de..d90dba8b2e8 100644
--- a/src/panfrost/encoder/pan_encoder.h
+++ b/src/panfrost/encoder/pan_encoder.h
@@ -66,4 +66,15 @@ panfrost_choose_hierarchy_mask(
         unsigned width, unsigned height,
         unsigned vertex_count, bool hierarchy);
 
+/* Stack sizes */
+
+unsigned
+panfrost_get_stack_shift(unsigned stack_size);
+
+unsigned
+panfrost_get_total_stack_size(
+                unsigned stack_shift,
+                unsigned threads_per_core,
+                unsigned core_count);
+
 #endif
diff --git a/src/panfrost/encoder/pan_scratch.c b/src/panfrost/encoder/pan_scratch.c
index 4a0561c7383..780dd539be6 100644
--- a/src/panfrost/encoder/pan_scratch.c
+++ b/src/panfrost/encoder/pan_scratch.c
@@ -77,3 +77,23 @@
  * There are other valid characterisations of this formula, of course, but this
  * is computationally simple, so good enough for our purposes.
  */
+
+/* Computes log_stack_size = ceil(log2(max(s, 256))) - 4 */
+
+unsigned
+panfrost_get_stack_shift(unsigned stack_size)
+{
+        return util_logbase2_ceil(MAX2(stack_size, 256)) - 4;
+}
+
+/* Computes the aligned stack size given the shift and thread count */
+
+unsigned
+panfrost_get_total_stack_size(
+                unsigned stack_shift,
+                unsigned threads_per_core,
+                unsigned core_count)
+{
+        unsigned stack_size = 1 << (stack_shift + 4);
+        return stack_size * threads_per_core * core_count;
+}