readd movedim

Padarn · Padarn · commit c9a3419af00a · 2021-10-23T07:58:23.000Z
diff --git a/functorch/csrc/BatchRulesViews.cpp b/functorch/csrc/BatchRulesViews.cpp
@@ -479,6 +479,7 @@ 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
 } 
 
 }}

Original file line number	Diff line number	Diff line change
`@@ -479,6 +479,7 @@ TORCH_LIBRARY_IMPL(aten, FT_BATCHED_KEY, m) {`
`479`	`479`	`VMAP_SUPPORT("expand", expand_batch_rule);`
`480`	`480`	`VMAP_SUPPORT("unfold", unfold_batch_rule);`
`481`	`481`	`VMAP_SUPPORT("movedim.intlist", movedim_batch_rule);`
	`482`	`+ m.impl("movedim.int", static_cast<Tensor(*)(const Tensor&,int64_t,int64_t)>(native::movedim)); // composite wrt autograd`
`482`	`483`	`}`
`483`	`484`
`484`	`485`	`}}`