Why dose TorchDispatchMode miss aten::_index_put_impl_

username: Cookie_Yang
cooked: <p>As the picture bellow, when I print my op’s call stack, I can see "aten::<em>index_put_impl</em> " in line 511.</p>
<p><div class="lightbox-wrapper"><a class="lightbox" href="https://discuss.pytorch.org/uploads/default/original/3X/f/a/fa85623e202f4c6665e11f8cd7dc6fc19706c5dc.png" data-download-href="https://discuss.pytorch.org/uploads/default/fa85623e202f4c6665e11f8cd7dc6fc19706c5dc" title="image"><img src="https://discuss.pytorch.org/uploads/default/optimized/3X/f/a/fa85623e202f4c6665e11f8cd7dc6fc19706c5dc_2_690x197.png" alt="image" data-base62-sha1="zKd8K0vnyc4E5iLUE3Pxct7b6by" width="690" height="197" srcset="https://discuss.pytorch.org/uploads/default/optimized/3X/f/a/fa85623e202f4c6665e11f8cd7dc6fc19706c5dc_2_690x197.png, https://discuss.pytorch.org/uploads/default/optimized/3X/f/a/fa85623e202f4c6665e11f8cd7dc6fc19706c5dc_2_1035x295.png 1.5x, https://discuss.pytorch.org/uploads/default/original/3X/f/a/fa85623e202f4c6665e11f8cd7dc6fc19706c5dc.png 2x" data-dominant-color="EB9070"><div class="meta"><svg class="fa d-icon d-icon-far-image svg-icon" aria-hidden="true"><use href="#far-image"></use></svg><span class="filename">image</span><span class="informations">1078×308 61.5 KB</span><svg class="fa d-icon d-icon-discourse-expand svg-icon" aria-hidden="true"><use href="#discourse-expand"></use></svg></div></a></div></p>
<p>But I cann’t capture "aten::<em>index_put_impl</em> " when I using <strong>TorchDispatchMode</strong>. What is the reason for this? TorchDispatchMode should capture atens which call actual kernel.</p>
<p>May related owner <a class="mention" href="/u/ezyang">@ezyang</a></p>
<aside class="onebox discoursetopic" data-onebox-src="https://dev-discuss.pytorch.org/t/what-and-why-is-torch-dispatch/557">
  <header class="source">
      <img src="https://canada1.discourse-cdn.com/flex036/uploads/pytorch1/optimized/1X/6cd7da56682d360e2c6006ff3e31eb250c5a8675_2_32x32.png" class="site-icon" width="32" height="32">

      <a href="https://dev-discuss.pytorch.org/t/what-and-why-is-torch-dispatch/557" target="_blank" rel="noopener nofollow ugc" title="02:32AM - 10 March 2022">PyTorch Developer Mailing List – 10 Mar 22</a>
  </header>

  <article class="onebox-body">
    <div class="aspect-image" style="--aspect-ratio:538/500;"><img src="https://canada1.discourse-cdn.com/flex036/uploads/pytorch1/optimized/1X/1fcf73bf511d7faf8f5b6315e4b9127e41d14fcb_2_1024x950.jpeg" class="thumbnail" width="538" height="500"></div>

<div class="title-wrapper">
  <h3><a href="https://dev-discuss.pytorch.org/t/what-and-why-is-torch-dispatch/557" target="_blank" rel="noopener nofollow ugc">What (and Why) is __torch_dispatch__?</a></h3>
  <div class="topic-category">
      <span class="badge-wrapper bullet">
        <span class="badge-category-bg" style="background-color: #9EB83B;"></span>
        <span class="badge-category clear-badge">
          <span class="category-name">frontend API</span>
        </span>
      </span>
  </div>
</div>

  <p>With Alban Desmaison, Edward Yang, and Richard Zou.  You may have seen us mention __torch_dispatch__ in various places recently. For example:   Making a Flop Counter that counts backwards FLOPS [Building a tracing frontend to make it easy to use...</p>

  <p>
    <span class="label1">Reading time: 4 mins 🕑</span>
      <span class="label2">Likes: 26 ❤</span>
  </p>

  </article>

  <div class="onebox-metadata">
    
    
  </div>

  <div style="clear: both"></div>
</aside>

<p>Looking forward for your reply!</p>