Could you rerun your code with CUDA_LAUNCH_BLOCKING=1 python script.py args and post the stack trace here, please?
Also, could you post all arguments and input shapes, so that we can reproduce this issue?