From f9618ea866d09141f77ebfea5cc7de0b9d7797fd Mon Sep 17 00:00:00 2001
From: Manfei <41607353+ManfeiBai@users.noreply.github.com>
Date: Wed, 17 Jan 2024 14:00:32 -0800
Subject: [PATCH] Update aten_xla_type.cpp

---
 torch_xla/csrc/aten_xla_type.cpp | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/torch_xla/csrc/aten_xla_type.cpp b/torch_xla/csrc/aten_xla_type.cpp
index 18a14cdde0b..e506d2c4b7b 100644
--- a/torch_xla/csrc/aten_xla_type.cpp
+++ b/torch_xla/csrc/aten_xla_type.cpp
@@ -2648,6 +2648,14 @@ at::Tensor scatter_reduce_helper(const at::Tensor& self, int64_t dim,
   }
 }
 
+at::Tensor XLANativeFunctions::scan(const at::Tensor& self,
+                                    const Callable f,
+                                    const at::Tensor& init,
+                                    const at::Tensor& xs) {
+  TORCH_LAZY_FN_COUNTER_TIMED_TRACING("xla::");
+  return bridge::AtenFromXlaTensor(tensor_methods::scan(f, init, xs));
+}
+
 at::Tensor XLANativeFunctions::scatter(const at::Tensor& self, int64_t dim,
                                        const at::Tensor& index,
                                        const at::Tensor& src) {