search

ROCm / rocAL

The AMD rocAL is designed to efficiently decode and process images and videos from a variety of storage formats and modify them through a processing graph programmable by the user.
https://rocm.docs.amd.com/projects/rocAL/en/develop/
MIT License
12 stars 14 forks source link

Fix external source reader pybind call #199

Closed swetha097 closed 2 months ago

swetha097 commented 3 months ago

This PR fixes the error with pybind call for external source reader

LakshmiKumar23 commented 3 months ago

@swetha097 in the python external_source_reader.py, please add an option to the user to pass CPU/GPU. Please try the app also. It hangs for me

swetha097 commented 3 months ago

@swetha097 in the python external_source_reader.py, please add an option to the user to pass CPU/GPU. Please try the app also. It hangs for me

Added option to pass cpu/gpu and batch_size

swetha097 commented 3 months ago

Addressed the comment @LakshmiKumar23

sbavasab commented 3 months ago

Is this supposed to be merged after last_batch_policy PR?

No this is an independent PR.