Rerun the code via CUDA_LAUNCH_BLOCKING=1 python script.py args as described in the error message or on the CPU to get the stacktrace pointing to the line of code causing the issue.
Often these asserts are triggered by e.g. an invalid indexing operation i