[functorch] [vmap] tests fail when `_set_vmap_fallback_enabled(False)`.
Labels: triaged, module: functorch

### 🐛 Describe the bug

With the following patch
```patch
diff --git a/test/functorch/test_vmap.py b/test/functorch/test_vmap.py
index a00612fdf5..f63651d9d4 100644
--- a/test/functorch/test_vmap.py
+++ b/test/functorch/test_vmap.py
@@ -58,6 +58,8 @@ from torch._functorch.make_functional import functional_init_with_buffers
 from torch.testing._internal.autograd_function_db import autograd_function_db
 from torch._functorch.vmap import restore_vmap
 
+torch._C._functorch._set_vmap_fallback_enabled(False)
+
 FALLBACK_REGEX = 'There is a performance drop'
 ```

A lot of tests fail

<details>

<summary> Failed Tests </summary>

```
============================================================================ short test summary info ============================================================================
FAILED test/functorch/test_vmap.py::TestVmapAPI::test_fallback_does_not_warn_by_default - RuntimeError: aten::_test_functorch_fallback hit the vmap fallback which is currentl...
FAILED test/functorch/test_vmap.py::TestVmapAPI::test_fallback_warning - RuntimeError: aten::_test_functorch_fallback hit the vmap fallback which is currently disabled
FAILED test/functorch/test_vmap.py::TestVmapAPI::test_fallback_zero_dim - AssertionError: "The fallback path does not support vmap over dims of size 0" does not match "aten::...
FAILED test/functorch/test_vmap.py::TestVmapAPI::test_inplace_fallback_nary_same_levels - RuntimeError: aten::atan2_ hit the vmap fallback which is currently disabled
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_linalg_eigh_cpu - RuntimeError: aten::linalg_eigh hit the vmap fallback which is currently disabled
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive__segment_reduce_lengths_cpu_float32 - RuntimeError: aten::segment_reduce hit the vmap fal...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive__segment_reduce_offsets_cpu_float32 - RuntimeError: aten::segment_reduce hit the vmap fal...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive__upsample_bilinear2d_aa_cpu_float32 - RuntimeError: aten::_upsample_bilinear2d_aa.vec hit...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_abs_cpu_float32 - RuntimeError: aten::absolute_ hit the vmap fallback which is currently ...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_acos_cpu_float32 - RuntimeError: aten::arccos_ hit the vmap fallback which is currently d...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_acosh_cpu_float32 - RuntimeError: aten::arccosh_ hit the vmap fallback which is currently...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_addbmm_cpu_float32 - RuntimeError: aten::addbmm_ hit the vmap fallback which is currently...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_addmm_cpu_float32 - RuntimeError: aten::addmm_ hit the vmap fallback which is currently d...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_addmm_decomposed_cpu_float32 - RuntimeError: aten::addmm_ hit the vmap fallback which is ...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_addmv_cpu_float32 - RuntimeError: aten::addmv_ hit the vmap fallback which is currently d...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_addr_cpu_float32 - RuntimeError: aten::addr_ hit the vmap fallback which is currently dis...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_argwhere_cpu_float32 - RuntimeError: aten::argwhere hit the vmap fallback which is curren...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_asin_cpu_float32 - RuntimeError: aten::arcsin_ hit the vmap fallback which is currently d...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_asinh_cpu_float32 - RuntimeError: aten::arcsinh_ hit the vmap fallback which is currently...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_atan2_cpu_float32 - RuntimeError: aten::atan2_ hit the vmap fallback which is currently d...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_atan_cpu_float32 - RuntimeError: aten::arctan_ hit the vmap fallback which is currently d...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_atanh_cpu_float32 - RuntimeError: aten::arctanh_ hit the vmap fallback which is currently...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_baddbmm_cpu_float32 - RuntimeError: aten::baddbmm_ hit the vmap fallback which is current...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_bincount_cpu_int64 - RuntimeError: aten::bincount hit the vmap fallback which is currentl...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_bucketize_cpu_float32 - RuntimeError: aten::bucketize.Tensor hit the vmap fallback which ...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_chalf_cpu_float32 - RuntimeError: aten::chalf hit the vmap fallback which is currently di...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_conj_physical_cpu_float32 - RuntimeError: aten::conj_physical_ hit the vmap fallback whic...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_count_nonzero_cpu_float32 - RuntimeError: aten::count_nonzero hit the vmap fallback which...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_cumprod_cpu_float32 - RuntimeError: aten::cumprod_ hit the vmap fallback which is current...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_cumsum_cpu_float32 - RuntimeError: aten::cumsum_ hit the vmap fallback which is currently...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_diagflat_cpu_float32 - RuntimeError: aten::diagflat hit the vmap fallback which is curren...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_div_floor_rounding_cpu_float32 - RuntimeError: aten::div_.Tensor_mode hit the vmap fallba...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_div_trunc_rounding_cpu_float32 - RuntimeError: aten::div_.Tensor_mode hit the vmap fallba...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_fft_ihfft2_cpu_float32 - RuntimeError: aten::fft_ihfft2 hit the vmap fallback which is cu...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_fft_ihfftn_cpu_float32 - RuntimeError: aten::fft_ihfftn hit the vmap fallback which is cu...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_fill_cpu_float32 - RuntimeError: aten::fill.Scalar hit the vmap fallback which is current...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_floor_divide_cpu_float32 - RuntimeError: aten::floor_divide_.Tensor hit the vmap fallback...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_fmod_cpu_float32 - RuntimeError: aten::fmod_.Tensor hit the vmap fallback which is curren...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_gcd_cpu_int64 - RuntimeError: aten::gcd_ hit the vmap fallback which is currently disabled
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_heaviside_cpu_float32 - RuntimeError: aten::heaviside_ hit the vmap fallback which is cur...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_histc_cpu_float32 - RuntimeError: aten::histc hit the vmap fallback which is currently di...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_histogram_cpu_float32 - RuntimeError: aten::histogram.bin_ct hit the vmap fallback which ...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_hypot_cpu_float32 - RuntimeError: aten::hypot_ hit the vmap fallback which is currently d...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_igamma_cpu_float32 - RuntimeError: aten::igamma_ hit the vmap fallback which is currently...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_igammac_cpu_float32 - RuntimeError: aten::igammac_ hit the vmap fallback which is current...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_index_add_cpu_float32 - RuntimeError: aten::index_add_ hit the vmap fallback which is cur...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_index_copy_cpu_float32 - RuntimeError: aten::index_copy_ hit the vmap fallback which is c...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_index_reduce_cpu_float32 - RuntimeError: aten::index_reduce hit the vmap fallback which i...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_isclose_cpu_float32 - RuntimeError: aten::isclose hit the vmap fallback which is currentl...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_isin_cpu_float32 - RuntimeError: aten::isin.Tensor_Tensor hit the vmap fallback which is ...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_istft_cpu_complex64 - RuntimeError: aten::istft hit the vmap fallback which is currently ...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_lcm_cpu_int64 - RuntimeError: aten::lcm_ hit the vmap fallback which is currently disabled
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_ldexp_cpu_float32 - RuntimeError: aten::ldexp_ hit the vmap fallback which is currently d...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_lerp_cpu_float32 - RuntimeError: aten::lerp_.Scalar hit the vmap fallback which is curren...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_linalg_ldl_solve_cpu_float32 - RuntimeError: aten::linalg_ldl_solve hit the vmap fallback...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_linalg_lu_cpu_float32 - RuntimeError: aten::linalg_lu hit the vmap fallback which is curr...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_linalg_tensorsolve_cpu_float32 - RuntimeError: aten::linalg_tensorsolve hit the vmap fall...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_lu_solve_cpu_float32 - RuntimeError: aten::lu_solve hit the vmap fallback which is curren...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_lu_unpack_cpu_float32 - RuntimeError: aten::lu_unpack hit the vmap fallback which is curr...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_masked_fill_cpu_float32 - RuntimeError: aten::masked_fill.Tensor hit the vmap fallback wh...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_masked_scatter_cpu_float32 - RuntimeError: aten::masked_scatter hit the vmap fallback whi...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_matrix_exp_cpu_float32 - RuntimeError: aten::matrix_exp hit the vmap fallback which is cu...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_nanquantile_cpu_float32 - RuntimeError: aten::nanquantile.scalar hit the vmap fallback wh...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_native_dropout_backward_cpu_float32 - RuntimeError: aten::native_dropout_backward hit the...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_neg_cpu_float32 - RuntimeError: aten::negative_ hit the vmap fallback which is currently ...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_nextafter_cpu_float32 - RuntimeError: aten::nextafter_ hit the vmap fallback which is cur...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_nn_functional_bilinear_cpu_float32 - RuntimeError: aten::bilinear hit the vmap fallback w...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_nn_functional_ctc_loss_cpu_float32 - RuntimeError: aten::ctc_loss.Tensor hit the vmap fal...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_nn_functional_huber_loss_cpu_float32 - RuntimeError: aten::huber_loss hit the vmap fallba...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_nn_functional_kl_div_cpu_float32 - RuntimeError: aten::kl_div hit the vmap fallback which...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_nn_functional_margin_ranking_loss_cpu_float32 - RuntimeError: aten::margin_ranking_loss h...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_nn_functional_max_pool1d_cpu_float32 - RuntimeError: aten::max_pool1d hit the vmap fallba...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_nn_functional_max_pool3d_cpu_float32 - RuntimeError: aten::max_pool3d_with_indices hit th...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_nn_functional_max_unpool1d_cpu_float32 - RuntimeError: aten::max_unpool2d hit the vmap fa...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_nn_functional_max_unpool1d_grad_cpu_float32 - RuntimeError: aten::max_unpool2d hit the vm...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_nn_functional_max_unpool2d_cpu_float32 - RuntimeError: aten::max_unpool2d hit the vmap fa...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_nn_functional_max_unpool2d_grad_cpu_float32 - RuntimeError: aten::max_unpool2d hit the vm...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_nn_functional_max_unpool3d_cpu_float32 - RuntimeError: aten::max_unpool3d hit the vmap fa...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_nn_functional_max_unpool3d_grad_cpu_float32 - RuntimeError: aten::max_unpool3d hit the vm...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_nn_functional_multi_margin_loss_cpu_float32 - RuntimeError: aten::multi_margin_loss hit t...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_nn_functional_multilabel_margin_loss_cpu_float32 - RuntimeError: aten::multilabel_margin_...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_nn_functional_pdist_cpu_float32 - RuntimeError: aten::pdist hit the vmap fallback which i...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_nn_functional_smooth_l1_loss_cpu_float32 - RuntimeError: aten::smooth_l1_loss hit the vma...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_nn_functional_soft_margin_loss_cpu_float32 - RuntimeError: aten::soft_margin_loss hit the...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_nn_functional_triplet_margin_loss_cpu_float32 - RuntimeError: aten::triplet_margin_loss h...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_ormqr_cpu_float32 - RuntimeError: aten::ormqr hit the vmap fallback which is currently di...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_polygamma_polygamma_n_0_cpu_float32 - RuntimeError: aten::polygamma_ hit the vmap fallbac...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_polygamma_polygamma_n_1_cpu_float32 - RuntimeError: aten::polygamma_ hit the vmap fallbac...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_polygamma_polygamma_n_2_cpu_float32 - RuntimeError: aten::polygamma_ hit the vmap fallbac...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_polygamma_polygamma_n_3_cpu_float32 - RuntimeError: aten::polygamma_ hit the vmap fallbac...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_polygamma_polygamma_n_4_cpu_float32 - RuntimeError: aten::polygamma_ hit the vmap fallbac...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_pow_cpu_float32 - RuntimeError: aten::pow_.Tensor hit the vmap fallback which is currentl...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_put_cpu_float32 - RuntimeError: aten::put hit the vmap fallback which is currently disabled
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_quantile_cpu_float32 - RuntimeError: aten::quantile.scalar hit the vmap fallback which is...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_remainder_cpu_float32 - RuntimeError: aten::remainder_.Tensor hit the vmap fallback which...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_renorm_cpu_float32 - RuntimeError: aten::renorm hit the vmap fallback which is currently ...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_scatter_add_cpu_float32 - RuntimeError: aten::scatter_add_ hit the vmap fallback which is...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_scatter_cpu_float32 - RuntimeError: aten::scatter_.src hit the vmap fallback which is cur...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_scatter_reduce_amax_cpu_float32 - RuntimeError: aten::scatter_reduce.two hit the vmap fal...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_scatter_reduce_amin_cpu_float32 - RuntimeError: aten::scatter_reduce.two hit the vmap fal...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_scatter_reduce_mean_cpu_float32 - RuntimeError: aten::scatter_reduce.two hit the vmap fal...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_scatter_reduce_prod_cpu_float32 - RuntimeError: aten::scatter_reduce.two hit the vmap fal...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_scatter_reduce_sum_cpu_float32 - RuntimeError: aten::scatter_reduce.two hit the vmap fall...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_special_airy_ai_cpu_float32 - RuntimeError: aten::special_airy_ai hit the vmap fallback w...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_special_bessel_j0_cpu_float32 - RuntimeError: aten::special_bessel_j0 hit the vmap fallba...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_special_bessel_j1_cpu_float32 - RuntimeError: aten::special_bessel_j1 hit the vmap fallba...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_special_bessel_y0_cpu_float32 - RuntimeError: aten::special_bessel_y0 hit the vmap fallba...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_special_bessel_y1_cpu_float32 - RuntimeError: aten::special_bessel_y1 hit the vmap fallba...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_special_chebyshev_polynomial_t_cpu_float32 - RuntimeError: aten::special_chebyshev_polyno...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_special_chebyshev_polynomial_u_cpu_float32 - RuntimeError: aten::special_chebyshev_polyno...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_special_hermite_polynomial_h_cpu_float32 - RuntimeError: aten::special_hermite_polynomial...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_special_hermite_polynomial_he_cpu_float32 - RuntimeError: aten::special_hermite_polynomia...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_special_laguerre_polynomial_l_cpu_float32 - RuntimeError: aten::special_laguerre_polynomi...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_special_log_ndtr_cpu_float32 - RuntimeError: aten::special_log_ndtr hit the vmap fallback...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_special_modified_bessel_i0_cpu_float32 - RuntimeError: aten::special_modified_bessel_i0 h...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_special_modified_bessel_i1_cpu_float32 - RuntimeError: aten::special_modified_bessel_i1 h...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_special_modified_bessel_k0_cpu_float32 - RuntimeError: aten::special_modified_bessel_k0 h...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_special_modified_bessel_k1_cpu_float32 - RuntimeError: aten::special_modified_bessel_k1 h...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_special_scaled_modified_bessel_k0_cpu_float32 - RuntimeError: aten::special_scaled_modifi...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_special_scaled_modified_bessel_k1_cpu_float32 - RuntimeError: aten::special_scaled_modifi...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_special_spherical_bessel_j0_cpu_float32 - RuntimeError: aten::special_spherical_bessel_j0...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_square_cpu_float32 - RuntimeError: aten::square_ hit the vmap fallback which is currently...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_stft_cpu_float32 - RuntimeError: aten::stft hit the vmap fallback which is currently disa...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_sub_cpu_float32 - RuntimeError: aten::subtract_.Tensor hit the vmap fallback which is cur...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_take_cpu_float32 - RuntimeError: aten::take hit the vmap fallback which is currently disa...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_trunc_cpu_float32 - RuntimeError: aten::fix_ hit the vmap fallback which is currently dis...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_exhaustive_xlogy_cpu_float32 - RuntimeError: aten::xlogy_.Tensor hit the vmap fallback which is curr...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_linalg_failure_1D_input_linalg_eigh_cpu_float32 - AssertionError: "dimension" does not match "aten::...
FAILED test/functorch/test_vmap.py::TestVmapOperatorsOpInfoCPU::test_vmap_linalg_failure_1D_input_linalg_lu_cpu_float32 - AssertionError: "dimension" does not match "aten::li...
==================================================== 129 failed, 1581 passed, 77 skipped, 226 xfailed in 1513.32s (0:25:13) ===================================================
```
</details>

More context: https://github.com/pytorch/pytorch/pull/98328#discussion_r1157628714

### Versions

master

cc @zou3519 @Chillee @samdow @soumith @janeyx99