viable/strict/1789588240: Move the miopen_batch_norm decomposition to the core table (#197160)

发生了什么

DTensor's sharding propagator has no strategy for aten.miopen_batch_norm , so F.batch_norm on a sharded 4D input fails on ROCm with "Operator aten.miopen_batch_norm.default does not have a sharding strategy registered". CUDA never hits the equivalent because aten.cudnn_batch_norm has a decomposition in torch/_decomp , which DTensor's DecompShardingStrategy falls back to. The ROCm decomposition existed only in torch/…

摘要按规则整理自下方来源原文

来源