radeonsi: add reference count to si_compute

[mesa.git] / src / gallium / drivers / radeonsi / si_compute.h

diff --git a/src/gallium/drivers/radeonsi/si_compute.h b/src/gallium/drivers/radeonsi/si_compute.h
index 268817b23a6e6ed9e5a207d7710b3663b769e079..c19b701fc710a9ce222aad49047a6106f9fcc523 100644 (file)
--- a/src/gallium/drivers/radeonsi/si_compute.h
+++ b/src/gallium/drivers/radeonsi/si_compute.h
@@ -24,11 +24,14 @@
 #ifndef SI_COMPUTE_H
 #define SI_COMPUTE_H
 
+#include "util/u_inlines.h"
+
 #include "si_shader.h"
 
 #define MAX_GLOBAL_BUFFERS 22
 
 struct si_compute {
+       struct pipe_reference reference;
        struct si_screen *screen;
        struct tgsi_token *tokens;
        struct util_queue_fence ready;
@@ -53,4 +56,15 @@ struct si_compute {
        unsigned uses_bindless_images:1;
 };
 
+void si_destroy_compute(struct si_compute *program);
+
+static inline void
+si_compute_reference(struct si_compute **dst, struct si_compute *src)
+{
+       if (pipe_reference(&(*dst)->reference, &src->reference))
+               si_destroy_compute(*dst);
+
+       *dst = src;
+}
+
 #endif /* SI_COMPUTE_H */