trunk/9455d3d34559799d09196341ca545d946610eb2e: Teach min-cut about online softmax prep (#185783)
🔒
https://github.com
«The online softmax preparation pattern computes softmax max and sum as a
multi-output prim. When this prim appears in an AOTAutograd joint graph, the
min-cut partitioner previously did not know that it was a recomputable...»
Automatische Weiterleitung...
1.5s