rohany / taco

The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs
http://tensor-compiler.org
Other
6 stars 3 forks source link

communicate: individual regions should allow for untrack, rather than the task as a whole #101

Open rohany opened 2 years ago

rohany commented 2 years ago

Marking the whole task as untrack is a bit heavy handed, as this can result in the runtime collecting instances that we don't want collected. Instead, communicate should have a marker on it that says if a particular communication operation should allow for collection.