ac: allow control flow statements in NIR callbacks

[mesa.git] / src / amd / common / ac_llvm_build.h

diff --git a/src/amd/common/ac_llvm_build.h b/src/amd/common/ac_llvm_build.h
index 8fcede66fb2b4997179d25515ff6fe94ce000969..e460d14e177684b1eceaf64e16943cfb795dd082 100644 (file)
--- a/src/amd/common/ac_llvm_build.h
+++ b/src/amd/common/ac_llvm_build.h
@@ -51,6 +51,12 @@ struct ac_llvm_flow;
 struct ac_llvm_compiler;
 enum ac_float_mode;
 
+struct ac_llvm_flow_state {
+       struct ac_llvm_flow *stack;
+       unsigned depth_max;
+       unsigned depth;
+};
+
 struct ac_llvm_context {
        LLVMContextRef context;
        LLVMModuleRef module;
@@ -93,9 +99,11 @@ struct ac_llvm_context {
        LLVMValueRef i1true;
        LLVMValueRef i1false;
 
-       struct ac_llvm_flow *flow;
-       unsigned flow_depth;
-       unsigned flow_depth_max;
+       /* Since ac_nir_translate makes a local copy of ac_llvm_context, there
+        * are two ac_llvm_contexts. Declare a pointer here, so that the control
+        * flow stack is shared by both ac_llvm_contexts.
+        */
+       struct ac_llvm_flow_state *flow;
 
        unsigned range_md_kind;
        unsigned invariant_load_md_kind;