frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

AOTI fallback ops: sort alphabetically (#150672)

Browse Source 
This is just a housekeeping task that makes the listed fallback op order match what's in the generated C shim files.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/150672
Approved by: https://github.com/desertfire
ghstack dependencies: #150671
This commit is contained in:
Benjamin Glass
2025-04-07 10:12:34 +00:00
committed by PyTorch MergeBot
parent 5e3c8214b5
commit f0abbabac1
1 changed files with 49 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

98
torchgen/aoti/fallback_ops.py
View File

@ -10,16 +10,55 @@
inductor_fallback_ops = {
"aten._adaptive_avg_pool2d_backward.default",
"aten._adaptive_avg_pool2d.default",
"aten._adaptive_avg_pool3d.default",
"aten._adaptive_avg_pool3d_backward.default",
"aten._adaptive_avg_pool3d.default",
"aten._addmm_activation.default",
"aten._cdist_backward.default",
"aten._cdist_forward.default",
"aten._cudnn_rnn.default",
"aten._dyn_quant_matmul_4bit.default",
"aten._dyn_quant_pack_4bit_weight.default",
"aten._efficient_attention_backward.default",
"aten._efficient_attention_forward.default",
"aten._efficientzerotensor.default",
"aten._embedding_bag_dense_backward.default",
"aten._embedding_bag_forward_only.default",
"aten._embedding_bag_per_sample_weights_backward.default",
"aten._embedding_bag.default",
"aten._fft_c2c.default",
"aten._fft_r2c.default",
"aten._flash_attention_backward.default",
"aten._flash_attention_forward.default",
"aten._fused_moving_avg_obs_fq_helper_functional.default",
"aten._fused_moving_avg_obs_fq_helper.default",
"aten._histogramdd_from_bin_cts.default",
"aten._int_mm.out",
"aten._pdist_backward.default",
"aten._pdist_forward.default",
"aten._scaled_dot_product_cudnn_attention_backward.default",
"aten._scaled_dot_product_cudnn_attention.default",
"aten._scaled_dot_product_efficient_attention_backward.default",
"aten._scaled_dot_product_efficient_attention.default",
"aten._scaled_dot_product_flash_attention_backward.default",
"aten._scaled_dot_product_flash_attention_for_cpu_backward.default",
"aten._scaled_dot_product_flash_attention_for_cpu.default",
"aten._scaled_dot_product_flash_attention.default",
"aten._scaled_dot_product_fused_attention_overrideable_backward.default",
"aten._scaled_dot_product_fused_attention_overrideable.default",
"aten._scaled_mm.default",
"aten._scaled_mm.out",
"aten._segment_reduce_backward.default",
"aten._thnn_fused_lstm_cell.default",
"aten._to_sparse.default",
"aten._trilinear.default",
"aten._weight_int8pack_mm.default",
"aten.adaptive_max_pool2d_backward.default",
"aten.adaptive_max_pool2d.default",
"aten.adaptive_max_pool3d.default",
"aten.adaptive_max_pool3d_backward.default",
"aten.adaptive_max_pool3d.default",
"aten.add.Scalar",
"aten.add.Tensor",
"aten.addbmm.default",
"aten._addmm_activation.default",
"aten.addmm.out",
"aten.addmv.default",
"aten.angle.default",
@ -33,57 +72,37 @@ inductor_fallback_ops = {
"aten.bmm.out",
"aten.bucketize.Tensor",
"aten.cat.default",
"aten._cdist_backward.default",
"aten._cdist_forward.default",
"aten.cholesky_inverse.default",
"aten.cholesky_solve.default",
"aten.convolution_backward.default",
"aten._cudnn_rnn.default",
"aten.convolution.default",
"aten.cummax.default",
"aten.cummin.default",
"aten.cumprod.default",
"aten.cumsum.default",
"aten._dyn_quant_matmul_4bit.default",
"aten._dyn_quant_pack_4bit_weight.default",
"aten._efficient_attention_backward.default",
"aten._efficient_attention_forward.default",
"aten._efficientzerotensor.default",
"aten._embedding_bag.default",
"aten._embedding_bag_dense_backward.default",
"aten._embedding_bag_forward_only.default",
"aten._embedding_bag_per_sample_weights_backward.default",
"aten.exponential.default",
"aten._fft_c2c.default",
"aten._fft_r2c.default",
"aten._flash_attention_backward.default",
"aten._flash_attention_forward.default",
"aten.fractional_max_pool2d_backward.default",
"aten.fractional_max_pool2d.default",
"aten.fractional_max_pool3d.default",
"aten.fractional_max_pool3d_backward.default",
"aten._fused_moving_avg_obs_fq_helper.default",
"aten._fused_moving_avg_obs_fq_helper_functional.default",
"aten.fractional_max_pool3d.default",
"aten.gcd.default",
"aten.geqrf.default",
"aten.grid_sampler_2d_backward.default",
"aten.histc.default",
"aten.histogram.bin_ct",
"aten._histogramdd_from_bin_cts.default",
"aten.index_put.default",
"aten.index_reduce.default",
"aten.index.Tensor",
"aten._int_mm.out",
"aten.kthvalue.default",
"aten.logcumsumexp.default",
"aten.lu_unpack.default",
"aten.masked_select.default",
"aten.masked_scatter.default",
"aten.masked_scatter_backward.default",
"aten.masked_scatter.default",
"aten.masked_select.default",
"aten.max_pool2d_with_indices_backward.default",
"aten.max_pool2d_with_indices.default",
"aten.max_pool3d_with_indices.default",
"aten.max_pool3d_with_indices_backward.default",
"aten.max_pool3d_with_indices.default",
"aten.max_unpool2d.default",
"aten.max_unpool3d.default",
"aten.median.default",
@ -93,11 +112,9 @@ inductor_fallback_ops = {
"aten.mul.Tensor",
"aten.nanmedian.default",
"aten.native_dropout.default",
"aten.normal_functional.default",
"aten.nonzero.default",
"aten.normal_functional.default",
"aten.ormqr.default",
"aten._pdist_backward.default",
"aten._pdist_forward.default",
"aten.polar.default",
"aten.pow.Scalar",
"aten.pow.Tensor_Scalar",
@ -106,8 +123,8 @@ inductor_fallback_ops = {
"aten.rand.generator",
"aten.randint.default",
"aten.randint.generator",
"aten.randint.low",
"aten.randint.low_out",
"aten.randint.low",
"aten.randn.default",
"aten.randn.generator",
"aten.randperm.default",
@ -117,36 +134,20 @@ inductor_fallback_ops = {
"aten.reshape.default",
"aten.resize_.default",
"aten.resize_as_.default",
"aten._scaled_dot_product_efficient_attention_backward.default",
"aten._scaled_dot_product_efficient_attention.default",
"aten._scaled_dot_product_flash_attention_backward.default",
"aten._scaled_dot_product_flash_attention.default",
"aten._scaled_dot_product_cudnn_attention_backward.default",
"aten._scaled_dot_product_cudnn_attention.default",
"aten._scaled_dot_product_flash_attention_for_cpu_backward.default",
"aten._scaled_dot_product_flash_attention_for_cpu.default",
"aten._scaled_dot_product_fused_attention_overrideable_backward.default",
"aten._scaled_dot_product_fused_attention_overrideable.default",
"aten._scaled_mm.default",
"aten._scaled_mm.out",
"aten.scatter_reduce.two_out",
"aten.scatter.src_out",
"aten.scatter.value_out",
"aten.searchsorted.Scalar",
"aten.searchsorted.Tensor",
"aten._segment_reduce_backward.default",
"aten.segment_reduce.default",
"aten.set_.source_Tensor",
"aten.slice.Tensor",
"aten.soft_margin_loss_backward.default",
"aten.sort.default",
"aten.sort.stable",
"aten._thnn_fused_lstm_cell.default",
"aten.topk.default",
"aten._to_sparse.default",
"aten.to_sparse.default",
"aten.topk.default",
"aten.triangular_solve.default",
"aten._trilinear.default",
"aten.uniform.default",
"aten.upsample_bicubic2d_backward.default",
"aten.upsample_linear1d_backward.default",
@ -154,5 +155,4 @@ inductor_fallback_ops = {
"aten.view_as_complex.default",
"aten.view_as_real.default",
"aten.view.dtype",
"aten._weight_int8pack_mm.default",
}