断言错误问题

[Alexisxty]

俺运行您这个demo测试的时候触发了这个报错，并且尝试了export CUDA_LAUNCH_BLOCKING=1，依然存在我问题

RuntimeError: CUDA error: device-side assert triggered
CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.
Compile with TORCH_USE_CUDA_DSA to enable device-side assertions.

十分感谢作者！
sincerely,
little cow horse

[erxianzhe]

我定位到了问题，#109 (comment)