@Alexey_Demyanchuk ,
In the beginning, I did work on 3090 machine. however, I couldn’t find out the resolution of the issue. I then tried it on 2080Ti machine.
After testing the code on 2080Ti, There is no issue on 2080Ti. the issue still exists on 3090 machine. It is the issue as the beginning.