Could you rerun your code via:
CUDA_LAUNCH_BLOCKING=1 python script.py args

and post the stack trace here, please?
Alternatively you could also run the code on the CPU and post the error message here.