Improve const-ness of module in run routines, and some of its methods.

PiperOrigin-RevId: 756032997

Author: dplassgit

xls/dslx/frontend/module.cc

@@ -130,8 +130,9 @@ std::vector<const AstNode*> Module::FindContained(const Span& target) const {
   return found;
 }
 
-std::optional<Function*> Module::GetFunction(std::string_view target_name) {
-  for (ModuleMember& member : top_) {
+std::optional<Function*> Module::GetFunction(
+    std::string_view target_name) const {
+  for (const ModuleMember& member : top_) {
     if (std::holds_alternative<Function*>(member)) {
       Function* f = std::get<Function*>(member);
       if (f->identifier() == target_name) {
@@ -142,8 +143,9 @@ std::optional<Function*> Module::GetFunction(std::string_view target_name) {
   return std::nullopt;
 }
 
-absl::StatusOr<TestFunction*> Module::GetTest(std::string_view target_name) {
-  for (ModuleMember& member : top_) {
+absl::StatusOr<TestFunction*> Module::GetTest(
+    std::string_view target_name) const {
+  for (const ModuleMember& member : top_) {
     if (std::holds_alternative<TestFunction*>(member)) {
       TestFunction* t = std::get<TestFunction*>(member);
       if (t->identifier() == target_name) {
@@ -155,8 +157,9 @@ absl::StatusOr<TestFunction*> Module::GetTest(std::string_view target_name) {
       "No test in module %s with name \"%s\"", name_, target_name));
 }
 
-absl::StatusOr<TestProc*> Module::GetTestProc(std::string_view target_name) {
-  for (ModuleMember& member : top_) {
+absl::StatusOr<TestProc*> Module::GetTestProc(
+    std::string_view target_name) const {
+  for (const ModuleMember& member : top_) {
     if (std::holds_alternative<TestProc*>(member)) {
       auto* t = std::get<TestProc*>(member);
       if (t->proc()->identifier() == target_name) {

xls/dslx/frontend/module.h

@@ -179,12 +179,12 @@ class Module : public AstNode {
                         T::GetDebugTypeName(), name_, target_name));
   }
 
-  std::optional<Function*> GetFunction(std::string_view target_name);
+  std::optional<Function*> GetFunction(std::string_view target_name) const;
 
   // Gets a test construct in this module with the given "target_name", or
   // returns a NotFoundError.
-  absl::StatusOr<TestFunction*> GetTest(std::string_view target_name);
-  absl::StatusOr<TestProc*> GetTestProc(std::string_view target_name);
+  absl::StatusOr<TestFunction*> GetTest(std::string_view target_name) const;
+  absl::StatusOr<TestProc*> GetTestProc(std::string_view target_name) const;
 
   absl::Span<ModuleMember const> top() const { return top_; }
   absl::Span<ModuleMember> top() { return absl::MakeSpan(top_); }

xls/dslx/run_routines/run_routines.cc

@@ -164,7 +164,7 @@ void HandleError(TestResultData& result, const absl::Status& status,
 };
 
 absl::Status RunDslxTestFunction(ImportData* import_data, TypeInfo* type_info,
-                                 Module* module, TestFunction* tf,
+                                 const Module* module, TestFunction* tf,
                                  const BytecodeInterpreterOptions& options) {
   auto cache = std::make_unique<BytecodeCache>();
   import_data->SetBytecodeCache(std::move(cache));
@@ -181,7 +181,7 @@ absl::Status RunDslxTestFunction(ImportData* import_data, TypeInfo* type_info,
 }
 
 absl::Status RunDslxTestProc(ImportData* import_data, TypeInfo* type_info,
-                             Module* module, TestProc* tp,
+                             const Module* module, TestProc* tp,
                              const BytecodeInterpreterOptions& options) {
   auto cache = std::make_unique<BytecodeCache>();
   import_data->SetBytecodeCache(std::move(cache));

xls/dslx/run_routines/run_routines.h

@@ -253,7 +253,7 @@ class DslxInterpreterTestRunner final : public AbstractTestRunner {
 class DslxInterpreterParsedTestRunner : public AbstractParsedTestRunner {
  public:
   DslxInterpreterParsedTestRunner(ImportData* import_data, TypeInfo* type_info,
-                                  Module* entry_module)
+                                  const Module* entry_module)
       : import_data_(import_data),
         type_info_(type_info),
         entry_module_(entry_module) {}
@@ -268,7 +268,7 @@ class DslxInterpreterParsedTestRunner : public AbstractParsedTestRunner {
  private:
   ImportData* import_data_;
   TypeInfo* type_info_;
-  Module* entry_module_;
+  const Module* entry_module_;
 };
 
 struct ParseAndProveResult {