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:
drewhk 2017-03-07 14:26:18 +01:00 committed by GitHub
parent 37610e43bb
commit 87d5b8f627
8 changed files with 301 additions and 66 deletions

View file

@ -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