What type of PR is this?
bug

What does this PR do / why do we need it:
Fix py_transforms gpu end-to-end testcases hang problem

Which issue(s) this PR fixes:

Fixes #I1HP4O

Special notes for your reviewers: