From c9a3419af00a4b83699ba5a39c6af6ce75b638ba Mon Sep 17 00:00:00 2001
From: Padarn Wilson <padarn.wilson@grabtaxi.com>
Date: Sat, 23 Oct 2021 07:58:23 +0000
Subject: [PATCH] readd movedim

---
 functorch/csrc/BatchRulesViews.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/functorch/csrc/BatchRulesViews.cpp b/functorch/csrc/BatchRulesViews.cpp
index de4f3830f..abd8e2602 100644
--- 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
 } 
 
 }}