Tags: orionr/pytorch
Tags
Backport transposes optimization to v0.3.0 (pytorch#3994) * Optimizer: optimize transposes in variety of circumstances (pytorch#3509) * Optimizer: Optimize transposes in variety of circumstances - No-op transposes - Consecutive transposes (fuse them) - Transposes into Gemm (fuse them into transA/transB parameter) * touch up out of date comment * Backporting optimizer changes
check for nvidia driver's sufficiency before checking for number of C… …UDA devices (pytorch#1156)
PreviousNext