Map data-type of fifos when cloning a block

PiperOrigin-RevId: 648877245
google · Jul 2, 2024 · 52703c5 · 52703c5
1 parent fb267c8
commit 52703c5
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/xls/ir/block.cc b/xls/ir/block.cc
@@ -748,10 +748,12 @@ absl::StatusOr<Block*> Block::Clone(
       XLS_ASSIGN_OR_RETURN(FifoInstantiation * fifo_inst,
                            inst->AsFifoInstantiation());
       XLS_ASSIGN_OR_RETURN(
-          instantiation_map[inst],
-          cloned_block->AddFifoInstantiation(
-              inst->name(), fifo_inst->fifo_config(), fifo_inst->data_type(),
-              fifo_inst->channel_name()));
+          Type * data_type,
+          target_package->MapTypeFromOtherPackage(fifo_inst->data_type()));
+      XLS_ASSIGN_OR_RETURN(instantiation_map[inst],
+                           cloned_block->AddFifoInstantiation(
+                               inst->name(), fifo_inst->fifo_config(),
+                               data_type, fifo_inst->channel_name()));
     }
   }