optimizer.py
9.31 KB
-
Fix the skipped algorithm finding in cached CUDNN convolution · a7a7e4fcSummary: This commit enforces the algorithm finding even if the backward of filter or data will not be executed. Otherwise, the empty algorithm will be encountered between two cached operation with the same arguments and input shape.
Ting PAN committed