From 9febb86967991e7d203ea71584e8d588b75370b4 Mon Sep 17 00:00:00 2001
From: Winnie Lin <wini@fb.com>
Date: Thu, 8 Sep 2022 20:19:21 -0700
Subject: [PATCH] [hotfix] fix build error from FuncTorch dispatch key

Reviewed By: ezyang

Differential Revision: D39374319

fbshipit-source-id: e1886e10bb3a5ed42f0a3127e3683f1b6787db80
---
 functorch/csrc/BatchRulesViews.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/functorch/csrc/BatchRulesViews.cpp b/functorch/csrc/BatchRulesViews.cpp
index 183efec19..b7e481f1e 100644
--- a/functorch/csrc/BatchRulesViews.cpp
+++ b/functorch/csrc/BatchRulesViews.cpp
@@ -505,7 +505,7 @@ std::tuple<Tensor, optional<int64_t>> diag_embed_batch_rule(const Tensor& self,
   return std::make_tuple(at::diag_embed(self_, offset, dim1, dim2), 0);
 }
 
-TORCH_LIBRARY_IMPL(aten, FT_BATCHED_KEY, m) {
+TORCH_LIBRARY_IMPL(aten, FuncTorchBatched, m) {
   VMAP_SUPPORT(diag, diag_batch_rule);
   VMAP_SUPPORT(chunk, chunk_batching_rule);
   m.impl("flatten.using_ints", static_cast<decltype(&ATEN_FN2(flatten, using_ints))>(native::flatten));