clean

pytorch · May 9, 2024 · 127ed7f · 127ed7f
1 parent 3ae7c97
commit 127ed7f
Showing 1 changed file with 0 additions and 83 deletions.
diff --git a/torch_xla/csrc/init_python_bindings.cpp b/torch_xla/csrc/init_python_bindings.cpp
@@ -706,87 +706,6 @@ runtime::ComputationClient::ComputationPtr CreateComputationFromProto(
       name, std::move(computation));
 }
 
-// // std::shared_ptr<XLAGraphExecutor::Async>
-// // XLAGraphExecutor::CompilationResult
-// // runtime::ComputationClient::ComputationPtr
-// // XLAGraphExecutor::saveComputation*
-// // std::vector<runtime::ComputationClient::ComputationPtr> 
-// // XLAGraphExecutor::CachedComputation 
-// runtime::ComputationClient::ComputationPtr GetXLAComputation(std::vector<XLATensorPtr> tensors,
-//     absl::Span<const std::string> devices, bool warm_up_cache_only = false) {
-//   tsl::profiler::TraceMe activity("GetXLAComputation", tsl::profiler::TraceMeLevel::kInfo);
-//   SyncTensorsConfig config;
-//   config.force_ltc_data = false;
-//   SyncTensorCollection coll = CollectSyncTensors(*tensors, config);
-//   if (coll.indices.empty()) {
-//     TensorCollectionBarrier(&coll);
-//     return nullptr;
-//   }
-//   DebugUtil::SaveTensorsGraphInfo("ScheduleSyncTensorsGraph", *tensors,
-//                                   &coll.indices);
-//   std::vector<torch::lazy::Value> ir_values;
-//   std::vector<torch::lazy::BackendDataPtr> tensor_data_vec;
-//   ExtractIRAndPrepareXlaData_(tensors, coll.config, coll.indices, ir_values,
-//                               tensor_data_vec);
-//   PostOrderData po_data = RunPostOrder(ir_values, &coll);
-//   coll.hash = torch::lazy::HashCombine(
-//       coll.hash, torch::lazy::Hash(po_data.parameter_sequence));
-//   if (GetAliasWithBufferDonorConfig()) {
-//     std::vector<size_t> buffer_donor_index =
-//         GetBufferDonorIndexFromUserConfig(po_data.parameters_data);
-//     if (buffer_donor_index.size() > 0) {
-//       // Do not include hash on a empty vector.
-//       coll.hash = torch::lazy::HashCombine(
-//           coll.hash, torch::lazy::Hash(buffer_donor_index));
-//     }
-//   }
-//   {
-//     // Auto-sharding configs
-//     coll.hash = torch::lazy::HashCombine(
-//         coll.hash, torch::lazy::MHash(ShardingUtil::GetAutoSharding()));
-//     coll.hash = torch::lazy::HashCombine(
-//         coll.hash,
-//         torch::lazy::StringHash(
-//             runtime::sys_util::GetEnvString("XLA_AUTO_SPMD_MESH", "").c_str()));
-//   }
-
-//   DebugUtil::SaveGraphHash(coll.hash);
-//   TF_VLOG(4) << "Parameter sequence graph hash "
-//              << torch::lazy::HashToString(coll.hash);
-
-//   std::pair<bool, std::shared_ptr<XLAGraphExecutor::Async>> cache_res =
-//       TryRunCachedSync(tensors, &coll, &po_data, tensor_data_vec,
-//                        warm_up_cache_only);
-//   if (cache_res.first) {
-//     // we have a cache hit, execution has been scheduled by TryRunCachedSync.
-//     return cache_res.second;
-//   }
-
-//   // CompilationResult compile_result =
-//   //     Compile(*tensors, devices, coll, &po_data, ir_values);
-
-//   // runtime::ComputationClient::ComputationPtr 
-//   // saveComputation* compile_result = std::move(
-//   //   Compile(*tensors, devices, coll, &po_data, ir_values).computation)
-//   XLAGraphExecutor::saveComputation* compile_result = Compile(*tensors, devices, coll, &po_data, ir_values).computation
-//   return compile_result
-
-//   // TORCH_LAZY_VALUE_METRIC("TensorsGraphSize", compile_result.emitted_nodes);
-//   // TF_VLOG(5) << "TensorsGraphSize=" << compile_result.emitted_nodes;
-//   // auto cached_computation = std::make_shared<CachedComputation>(
-//   //     std::move(compile_result.computation), compile_result.is_sharded);
-//   // GetComputationCache()->Add(coll.hash, cached_computation);
-
-//   // if (warm_up_cache_only) {
-//   //   return nullptr;
-//   // } else {
-//   //   return ScheduleSyncTensorsGraph(
-//   //       tensors, &coll, std::move(compile_result.parameters_data),
-//   //       compile_result.device.toString(), std::move(cached_computation),
-//   //       tensor_data_vec);
-//   // }
-// }
-
 xla::Shape GetTensorShape(const at::Tensor& tensor,
                           const std::string& device_str) {
   auto xtensor = bridge::TryGetXlaTensor(tensor);
@@ -966,8 +885,6 @@ class PyLoweringContext {
  public:
   PyLoweringContext() : PyLoweringContext(bridge::GetCurrentDevice()) {}
 
-  // PostOrderData po_data = RunPostOrder(ir_values, &coll);
-
   PyLoweringContext(torch::lazy::BackendDevice device)
       : lowering_ctx("PyLoweringContext", device) {}