title: CUDA error: device-side assert triggered CUDA kernel errors might be asynchronously reported at some other API call
body: <p>Just for study, I just tried to run the code below.</p>
  <p><a href="https://colab.research.google.com/drive/1dFC0FL-521m7CL_PSd8RLKq67jgTJVhL?usp=sharing" rel="nofollow noreferrer">https://colab.research.google.com/drive/1dFC0FL-521m7CL_PSd8RLKq67jgTJVhL?usp=sharing</a></p>
  <p>But there's an error &quot;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.&quot;</p>
  <p>Details are in the picture.</p>
  <p><img src="https://i.stack.imgur.com/BDmi9.png" alt="enter image description here" /></p>
  <p>What can I do to solve this problem?</p>