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/master/index.html
MIT License
10 stars 13 forks source link

Bug Fix - Tests fall back to CPU if CuPy unavailable #137

Closed LakshmiKumar23 closed 2 months ago

LakshmiKumar23 commented 2 months ago

Adding a try...except block to check for CuPy. Upgrading to rocm6.2 fails to build CuPy. This PR makes the generic and tf plugins fall back and use CPU is CuPy is not available. Tested decoder.py, tf_classification and tf_detection and unit_test.sh