tensorflow / tensorflow

An Open Source Machine Learning Framework for Everyone
https://tensorflow.org
Apache License 2.0
182.91k stars 73.92k forks source link

Move fuel consumption point to an earlier stage in the cudnn fused convolution rewriter #67155

Closed copybara-service[bot] closed 1 week ago

copybara-service[bot] commented 1 week ago

Move fuel consumption point to an earlier stage in the cudnn fused convolution rewriter

The call to ConsumeFuel happened after EnsureIfConvBiasActivation had already rewritten the convolution instruction. In order to effectively use ConsumeFuel for model bisection I'm moving the check to before any changes are made to the instruction.