Update TODOs in XLA to reference bug numbers.

This change replaces TODOs associated with me with references to tracking bugs for better management. One obsolete TODO in scatter.cc is also removed.

PiperOrigin-RevId: 900788150

Author: Google-ML-Automation

xla/backends/gpu/codegen/emitters/scatter.cc

@@ -601,7 +601,6 @@ void ScatterWithDistributedIndices::ComputeIndexing(
   }
 
   if (updates_map) {
-    // TODO (karupayun): Check number of dimensions.
     auto index_id = CreateSymbolExpr(0, kGpuGridDims, mlir_context);
     auto update_dim_loop = CreateSymbolExpr(1, kGpuGridDims, mlir_context);
     auto vector_id = CreateSymbolExpr(2, kGpuGridDims, mlir_context);

xla/hlo/analysis/indexing_map_serialization.cc

@@ -899,7 +899,7 @@ std::string ToString(const IndexingMap& indexing_map,
   symbol_names.reserve(range_names.size() + rt_names.size());
   symbol_names.append(range_names.begin(), range_names.end());
   symbol_names.append(rt_names.begin(), rt_names.end());
-  // TODO(karupayun): Do not use conversion here.
+  // TODO(b/446856305): Do not use conversion here.
   ss << ToString(SymbolicMapToAffineMap(indexing_map.GetSymbolicMap()),
                  dim_names, range_names, rt_names);
   if (dim_vars.empty() && range_vars.empty() && rt_vars.empty()) {

xla/hlo/analysis/indexing_map_test.cc

@@ -621,7 +621,7 @@ TEST_F(IndexingMapTest, ConstraintIntervalSimplification_Sum) {
     d0 mod 8 + 5 in [50, 54]
   )");
   EXPECT_TRUE(indexing_map.Simplify());
-  // TODO(karupayun): This should be infeasible, since d0 mod 8 should be in
+  // TODO: b/459357586 - This should be infeasible, since d0 mod 8 should be in
   // [0, 7].
   EXPECT_THAT(ToString(indexing_map), MatchIndexingString(R"(
                           (d0) -> (d0),
@@ -678,7 +678,7 @@ TEST_F(IndexingMapTest,
     s1 in [0, 2],
     d0 * 6 + s0 * 3 + s1 in [0, 598]
   )");
-  // TODO(karupayun): This should be simplified to
+  // TODO: b/459357586 - This should be simplified to
   // (d0)[s0, s1] -> (d0 * 6 + s0 * 3 + s1),
   // domain:
   // d0 in [0, 99],

xla/hlo/analysis/symbolic_expr.h

@@ -47,7 +47,7 @@ enum class SymbolicExprType {
   kMin,
   kVariable,
   kConstant,  // Constant should be the last type for the comparator.
-  // TODO(karupayun): Add kIn operator.
+  // TODO: b/459357586 - Add kIn operator.
   // kIn,  // 'var in [a, b]' .
 };