media: venus: Mark last capture buffer

[sfrench/cifs-2.6.git] / drivers / media / platform / qcom / venus / hfi.c

diff --git a/drivers/media/platform/qcom/venus/hfi.c b/drivers/media/platform/qcom/venus/hfi.c
index 3d8b1284d1f35386b598f2c41deff1fd1e572b98..a211eb93e0f924f6ec22a6eb6b86fdddaa8cd797 100644 (file)
--- a/drivers/media/platform/qcom/venus/hfi.c
+++ b/drivers/media/platform/qcom/venus/hfi.c
@@ -382,7 +382,7 @@ int hfi_session_unload_res(struct venus_inst *inst)
 }
 EXPORT_SYMBOL_GPL(hfi_session_unload_res);
 
-int hfi_session_flush(struct venus_inst *inst, u32 type)
+int hfi_session_flush(struct venus_inst *inst, u32 type, bool block)
 {
        const struct hfi_ops *ops = inst->core->ops;
        int ret;
@@ -393,9 +393,11 @@ int hfi_session_flush(struct venus_inst *inst, u32 type)
        if (ret)
                return ret;
 
-       ret = wait_session_msg(inst);
-       if (ret)
-               return ret;
+       if (block) {
+               ret = wait_session_msg(inst);
+               if (ret)
+                       return ret;
+       }
 
        return 0;
 }