Tags: Tianshu-Bao/pytorch
Tags
Scopes 0.3.1 backport (pytorch#5153) * Introduce scopes during tracing (pytorch#3016) * Fix segfault during ONNX export * Further fix to tracing scope (pytorch#4558) * Set missing temporary scope in callPySymbolicMethod * Use expected traces in all scope tests * Fix tracking of tracing scopes during ONNX pass (pytorch#4524) * Fix tracking of tracing scopes during ONNX pass * Use ResourceGuard to manage setting a temporary current scope in Graph * Add tests for ONNX pass scopes * Remove unused num_classes argument * Expose node scopeName to python (pytorch#4200) * Inherit JIT scopes when cloning only when it's correct It's correct only when the new graph owns the same scope tree as the original one. We can end up with dangling pointers otherwise. * Fixes after cherry-picking, still one test to go * Fix for last failing test after scope cherry-pick * Fix linting issue
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