anv: flag batch & instruction BOs for capture

[mesa.git] / src / util / register_allocate.h

diff --git a/src/util/register_allocate.h b/src/util/register_allocate.h
index 6abb4e04d0bbf93def6650fb7f84f2eb06b685be..7c40542641b111b05c7ce857bb1dcff9fcb53ef3 100644 (file)
--- a/src/util/register_allocate.h
+++ b/src/util/register_allocate.h
@@ -29,6 +29,7 @@
 #define REGISTER_ALLOCATE_H
 
 #include <stdbool.h>
+#include "util/bitset.h"
 
 #ifdef __cplusplus
 extern "C" {
@@ -74,6 +75,11 @@ void ra_set_finalize(struct ra_regs *regs, unsigned int **conflicts);
 struct ra_graph *ra_alloc_interference_graph(struct ra_regs *regs,
                                             unsigned int count);
 void ra_set_node_class(struct ra_graph *g, unsigned int n, unsigned int c);
+void ra_set_select_reg_callback(struct ra_graph *g,
+                                unsigned int (*callback)(struct ra_graph *g,
+                                                         BITSET_WORD *regs,
+                                                         void *data),
+                                void *data);
 void ra_add_node_interference(struct ra_graph *g,
                              unsigned int n1, unsigned int n2);
 /** @} */