diff --git a/third_party/nvfuser/csrc/ir_cloner.h b/third_party/nvfuser/csrc/ir_cloner.h
index 35016c3596e9..0e2ef3a58374 100644
--- a/third_party/nvfuser/csrc/ir_cloner.h
+++ b/third_party/nvfuser/csrc/ir_cloner.h
@@ -4,7 +4,9 @@
 #include <dispatch.h>
 #include <ir_builder.h>
 
+#include <tuple>
 #include <unordered_map>
+#include <unordered_set>
 #include <vector>
 
 namespace nvfuser {
@@ -28,6 +30,10 @@ class TORCH_CUDA_CU_API IrCloner {
 
   Statement* clone(const Statement* statement);
 
+  int64_t clone(int64_t x) {
+    return x;
+  }
+
   template <class T>
   T* clone(const T* node) {
     return node ? clone(node->template as<Statement>())->template as<T>()
@@ -35,9 +41,9 @@ class TORCH_CUDA_CU_API IrCloner {
   }
 
   template <class T>
-  std::vector<T*> clone(const std::vector<T*>& container) {
+  std::vector<T> clone(const std::vector<T>& container) {
     // NOLINTNEXTLINE(cppcoreguidelines-init-variables)
-    std::vector<T*> copy;
+    std::vector<T> copy;
     copy.reserve(container.size());
     for (auto p : container) {
       copy.push_back(clone(p));
@@ -45,6 +51,23 @@ class TORCH_CUDA_CU_API IrCloner {
     return copy;
   }
 
+  template <class T>
+  std::unordered_set<T> clone(const std::unordered_set<T>& container) {
+    std::unordered_set<T> copy;
+    copy.reserve(container.size());
+    for (auto p : container) {
+      copy.insert(clone(p));
+    }
+    return copy;
+  }
+
+  template <typename... Ts>
+  std::tuple<Ts...> clone(const std::tuple<Ts...>& tup) {
+    return std::apply(
+        [this](auto&... x) { return std::make_tuple<Ts...>(clone(x)...); },
+        tup);
+  }
+
   IrContainer* container() const {
     return ir_container_;
   }