Wip fix islands drewhk (#22475)
* Fixed island application * Switched to explicit island boundaries and fixed misapplied island tags * Added missing attributes for non-stage backed Sinks
This commit is contained in:
parent
37610e43bb
commit
87d5b8f627
8 changed files with 301 additions and 66 deletions
|
|
@ -52,8 +52,10 @@ abstract class SinkModule[-In, Mat](val shape: SinkShape[In]) extends AtomicModu
|
|||
*/
|
||||
def create(context: MaterializationContext): (AnyRef, Mat)
|
||||
|
||||
def attributes: Attributes
|
||||
|
||||
override def traversalBuilder: TraversalBuilder =
|
||||
LinearTraversalBuilder.fromModule(this).makeIsland(SinkModuleIslandTag)
|
||||
LinearTraversalBuilder.fromModule(this, attributes).makeIsland(SinkModuleIslandTag)
|
||||
|
||||
// This is okay since we the only caller of this method is right below.
|
||||
// TODO: Remove this, no longer needed
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue