Remove XLATensor from LookupCachedCompile

Author: alanwaketan

torch_xla/csrc/xla_graph_executor.cpp

@@ -850,7 +850,7 @@ XLAGraphExecutor::PostOrderData XLAGraphExecutor::RunPostOrder(
 }
 
 XLAGraphExecutor::ComputationCache::TypePtr
-XLAGraphExecutor::LookupCachedCompile(const std::vector<XLATensorPtr>& tensors,
+XLAGraphExecutor::LookupCachedCompile(
                                       const torch::lazy::hash_t& hash) {
   ComputationCache::TypePtr cached_computation =
       GetComputationCache()->Get(hash);
@@ -873,7 +873,7 @@ std::shared_ptr<XLAGraphExecutor::Async> XLAGraphExecutor::TryRunCachedSync(
     PostOrderData* po_data,
     const std::vector<torch::lazy::BackendDataPtr>& tensor_data_vec) {
   ComputationCache::TypePtr cached_computation =
-      LookupCachedCompile(*tensors, coll->hash);
+      LookupCachedCompile(coll->hash);
   if (cached_computation == nullptr) {
     return nullptr;
   }

torch_xla/csrc/xla_graph_executor.h

@@ -261,13 +261,12 @@ class XLAGraphExecutor : public torch::lazy::LazyGraphExecutor {
   PostOrderData RunPostOrder(const std::vector<torch::lazy::Value>& ir_values,
                              SyncTensorCollection* coll);
 
-  // We don't use the upstream LookupCachedCompile since we need XLATensorPtr and
+  // We don't use the upstream LookupCachedCompile since
   // our CachedComputation is different from upstream.
   ComputationCache::TypePtr LookupCachedCompile(
-      const std::vector<XLATensorPtr>& tensors,
       const torch::lazy::hash_t& hash);
 
-  // We don't use the upstream TryRunCachedSync since we need XLATensorPtr and
+  // We don't use the upstream TryRunCachedSync since
   // our CachedComputation is different from upstream.
   std::shared_ptr<Async> TryRunCachedSync(
       std::vector<XLATensorPtr>* tensors, SyncTensorCollection* coll,