habanalabs: release signal if collective wait was dropped

As in standard wait cs, we must release a signal fence once
a collective wait cs was dropped and not submitted.

Signed-off-by: Ofir Bitton <obitton@habana.ai>
Reviewed-by: Oded Gabbay <ogabbay@kernel.org>
Signed-off-by: Oded Gabbay <ogabbay@kernel.org>

diff --git a/drivers/misc/habanalabs/common/command_submission.c b/drivers/misc/habanalabs/common/command_submission.c
index 56dc18f0bc2709bb687b2183569eef5db61f4aea..536e09542e223de7f5eecc059a564ef29dfee671 100644 (file)
--- a/drivers/misc/habanalabs/common/command_submission.c
+++ b/drivers/misc/habanalabs/common/command_submission.c
@@ -300,9 +300,11 @@ static void cs_do_release(struct kref *ref)
 
        if (!cs->submitted) {
                /* In case the wait for signal CS was submitted, the put occurs
-                * in init_signal_wait_cs() right before hanging on the PQ.
+                * in init_signal_wait_cs() or collective_wait_init_cs()
+                * right before hanging on the PQ.
                 */
-               if (cs->type == CS_TYPE_WAIT)
+               if (cs->type == CS_TYPE_WAIT ||
+                               cs->type == CS_TYPE_COLLECTIVE_WAIT)
                        hl_fence_put(cs->signal_fence);
 
                goto out;