moved movedim decomposition to batchrulesstopdecomposition

Author: Chillee

functorch/csrc/BatchRulesStopDecomposition.cpp

@@ -72,6 +72,7 @@ TORCH_LIBRARY_IMPL(aten, FT_BATCHED_KEY, m) {
   OP_DECOMPOSE(mH);
   OP_DECOMPOSE2(min, other );
   OP_DECOMPOSE2(moveaxis, intlist);
+  OP_DECOMPOSE2(movedim, int);
   OP_DECOMPOSE(msort);
   OP_DECOMPOSE(mT);
   OP_DECOMPOSE2(multiply, Tensor );

functorch/csrc/BatchRulesViews.cpp

@@ -479,7 +479,6 @@ TORCH_LIBRARY_IMPL(aten, FT_BATCHED_KEY, m) {
   VMAP_SUPPORT("expand", expand_batch_rule);
   VMAP_SUPPORT("unfold", unfold_batch_rule);
   VMAP_SUPPORT("movedim.intlist", movedim_batch_rule);
-  m.impl("movedim.int", static_cast<Tensor(*)(const Tensor&,int64_t,int64_t)>(native::movedim)); // composite wrt autograd
-} 
+}
 
 }}