pytorch
d371a9f9 - Change ScatterGather kernel names on dtype dispatch. (#54498)

Commit
3 years ago
Change ScatterGather kernel names on dtype dispatch. (#54498) Summary: Changed `ScatterGather` kernel name when `dtype` dispatching to a more meaningful name than `"method_name"`. https://github.com/pytorch/pytorch/blob/2a53897114e984f5433e9e495cccb1417b86a8d3/aten/src/ATen/native/cpu/ScatterGatherKernel.cpp#L146-L148 https://github.com/pytorch/pytorch/blob/2a53897114e984f5433e9e495cccb1417b86a8d3/aten/src/ATen/native/cpu/ScatterGatherKernel.cpp#L241-L243 Maybe, a more specific name, based on who is calling (e.g. `gather_cpu_kernel`, `scatter_cpu_kernel`), would be better. Any thoughts? Pull Request resolved: https://github.com/pytorch/pytorch/pull/54498 Reviewed By: malfet Differential Revision: D27291514 Pulled By: bdhirsh fbshipit-source-id: 123b77296e685ee34031da661c78e201a10757db
Author
Parents
Loading