Modify the memory pool access for batch decode submission (#50)

[ROCm/rocjpeg commit: 7129a01f88]
2024-08-14 16:45:05 -04:00
parent 7eeba4c27a
commit 88741716d4
3 changed files with 93 additions and 30 deletions
@@ -172,6 +172,7 @@ RocJpegStatus RocJpegDecoder::Decode(RocJpegStreamHandle jpeg_stream_handle, con
            break;
    }

+    CHECK_ROCJPEG(jpeg_vaapi_decoder_.SetSurfaceAsIdle(current_surface_id));
    CHECK_HIP(hipStreamSynchronize(hip_stream_));
    return ROCJPEG_STATUS_SUCCESS;
 }
@@ -268,6 +269,7 @@ RocJpegStatus RocJpegDecoder::DecodeBatched(RocJpegStreamHandle *jpeg_streams, i
                default:
                    break;
            }
+            CHECK_ROCJPEG(jpeg_vaapi_decoder_.SetSurfaceAsIdle(current_surface_id));
        }

    }