As the picture bellow, when I print my op’s call stack, I can see "aten::index_put_impl " in line 511.
But I cann’t capture "aten::index_put_impl " when I using TorchDispatchMode. What is the reason for this? TorchDispatchMode should capture atens which call actual kernel.
May related owner @ezyang
Looking forward for your reply!