mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
b21a6ff639
Syncing nvfuser devel branch to upstream master. https://github.com/csarofeen/pytorch/ Code changes includes: - codegen improvements: 1. double support in expression evaluator - bug fixes: 1. dropout fix - rework RNG to support broadcasted dropout (Fixes #82784) 2. expand fix - Patch expand+reduction, expand+view, rework view analysis and guard - scheduler: 1. manual transpose schedule example 2. WIP transpose scheduler Commits that's in this PR from the devel branch: ``` b7435afcd22c917713c2f41a7237bc26e1183f14 Transpose scheduler, step 1 (#1854) 8a45dbf72034684eb8e18b1835b533e90b68f184 Add an example on how to manually schedule transpose (#1889) 83dbf56a9554b2efbd5416461d938fff477b0b27 Patch dropout fix (#1898) 69d3519a532250719b1aa8341b50e067b181b42d Expand+Reduction, Expand+View support, rework View analysis and guards (#1883) 15091c488e96343bdc49e3990acbf238a3b3da51 Rework RNG to correctly support broadcasted dropout (#1888) aafe2d048aaac596e503596a41303423619f3954 Make ExpressionEvaluator support Double (#1885) ``` RUN_TORCHBENCH: nvfuser Differential Revision: [D38657074](https://our.internmc.facebook.com/intern/diff/D38657074) Pull Request resolved: https://github.com/pytorch/pytorch/pull/83239 Approved by: https://github.com/davidberard98 |
||
|---|---|---|
| .. | ||
| batch_norm_channels_first.cpp | ||
| batch_norm_channels_first_backward.cpp | ||
| batch_norm_channels_last.cpp | ||
| batch_norm_channels_last_backward.cpp | ||
| bert.cpp | ||
| broadcast.cpp | ||
| CMakeLists.txt | ||
| gelu_backward.cpp | ||
| heuristic_cache.cpp | ||
| heuristic_lookup.cpp | ||
| instance_norm.cpp | ||
| layer_norm.cpp | ||
| layer_norm_backward.cpp | ||
| lstm_cell.cpp | ||
| main.cpp | ||
| reduction.cpp | ||
| rms_norm.cpp | ||
| rms_norm_backward.cpp | ||
| scale_bias_relu.cpp | ||
| shape_inference.cpp | ||
| softmax.cpp | ||
| softmax_backward.cpp | ||
| softmax_dropout.cpp | ||
| timm.cpp | ||
| transpose.cpp | ||
| utils.cpp | ||
| utils.h | ||