[Proc-scoped channels] Update scheduling options to properly map IO constraints to channel interfaces of proc-scoped channels.

Update custom schedule codegen to use proc-scoped channels.

PiperOrigin-RevId: 841947015

Author: dplassgit

xls/examples/BUILD

@@ -171,6 +171,9 @@ xls_dslx_opt_ir(
     name = "custom_schedule",
     srcs = ["custom_schedule.x"],
     dslx_top = "Accumulator",
+    ir_conv_args = {
+        "lower_to_proc_scoped_channels": "true",
+    },
 )
 
 xls_dslx_fmt_test(
@@ -196,11 +199,14 @@ xls_dslx_verilog(
         "streaming_channel_data_suffix": "_data",
         "streaming_channel_ready_suffix": "_ready",
         "streaming_channel_valid_suffix": "_valid",
-        "io_constraints": "custom_schedule__data_in:recv:custom_schedule__old_state:send:0:0,custom_schedule__data_in:recv:custom_schedule__data_out:send:4:4,custom_schedule__data_in:recv:custom_schedule__activate:recv:8:8",
+        "io_constraints": "data_in:recv:old_state:send:0:0,data_in:recv:data_out:send:4:4,data_in:recv:activate:recv:8:8",
         "worst_case_throughput": "9",
         "pipeline_stages": "9",
     },
     dslx_top = "Accumulator",
+    ir_conv_args = {
+        "lower_to_proc_scoped_channels": "true",
+    },
     library = ":custom_schedule",
     verilog_file = "custom_schedule.sv",
 )

xls/scheduling/BUILD

@@ -95,6 +95,7 @@ cc_library(
         "//xls/estimators/delay_model:delay_estimator",
         "//xls/estimators/delay_model:delay_estimators",
         "//xls/ir",
+        "//xls/ir:channel",
         "//xls/passes:optimization_pass",
         "//xls/tools:scheduling_options_flags_cc_proto",
         "@com_google_absl//absl/container:flat_hash_map",

xls/scheduling/scheduling_options.cc

@@ -30,6 +30,7 @@
 #include "xls/common/status/status_macros.h"
 #include "xls/estimators/delay_model/delay_estimator.h"
 #include "xls/estimators/delay_model/delay_estimators.h"
+#include "xls/ir/channel.h"
 #include "xls/ir/package.h"
 #include "xls/tools/scheduling_options_flags.pb.h"
 
@@ -152,6 +153,34 @@ absl::StatusOr<SchedulingOptions> OptionsFromFlagProto(
     for (const SchedulingConstraint& c : scheduling_options.constraints()) {
       if (std::holds_alternative<IOConstraint>(c)) {
         IOConstraint io_constr = std::get<IOConstraint>(c);
+        if (p->ChannelsAreProcScoped()) {
+          XLS_ASSIGN_OR_RETURN(Proc * top, p->GetTopAsProc());
+          if (!top->GetChannelInterface(
+                      io_constr.SourceChannel(),
+                      (io_constr.SourceDirection() == IODirection::kSend)
+                          ? ChannelDirection::kSend
+                          : ChannelDirection::kReceive)
+                   .ok()) {
+            return absl::InvalidArgumentError(absl::StrFormat(
+                "Invalid source channel name in IO constraint: %s; "
+                "this name did not correspond to any channel interface in the "
+                "top proc %s",
+                io_constr.SourceChannel(), top->name()));
+          }
+          if (!top->GetChannelInterface(
+                      io_constr.TargetChannel(),
+                      (io_constr.TargetDirection() == IODirection::kSend)
+                          ? ChannelDirection::kSend
+                          : ChannelDirection::kReceive)
+                   .ok()) {
+            return absl::InvalidArgumentError(
+                absl::StrFormat("Invalid channel name in IO constraint: %s; "
+                                "this name did not correspond to any channel "
+                                "interface in the top proc %s",
+                                io_constr.TargetChannel(), top->name()));
+          }
+          continue;
+        }
         if (!p->GetChannel(io_constr.SourceChannel()).ok()) {
           return absl::InvalidArgumentError(absl::StrFormat(
               "Invalid channel name in IO constraint: %s; "