pytorch

mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00

History

Ke Wen 3090667cf9 [pipelining] pipeline() taking microbatch as example input (#128163 ) Changed the API of `pipeline()` to take microbatch instead of full batch as example args. Main purpose is to: - make this API more atomic; - decouple tracing frontend from runtime info like `num_chunks`. Side effects: - Creates opportunity for varying `num_chunks` of schedules with the same `pipe` object. - User has to create example microbatch input. - Chunk spec stuff are now all moved to runtime side. Pull Request resolved: https://github.com/pytorch/pytorch/pull/128163 Approved by: https://github.com/H-Huang		2024-06-07 15:51:53 +00:00
..
caffe2	Enable UFMT on a bunch of low traffic Python files outside of main files (#106052 )	2023-07-27 01:01:17 +00:00
cpp	Update Jinja to 3.1.3 (#124976 )	2024-04-26 02:57:55 +00:00
source	[pipelining] pipeline() taking microbatch as example input (#128163 )	2024-06-07 15:51:53 +00:00
.gitignore
libtorch.rst	Replace master with main in links and docs/conf.py (#100176 )	2023-05-02 18:20:32 +00:00
make.bat
Makefile	Refactor torch.onnx documentation (#108379 )	2023-09-08 18:23:48 +00:00
README.md
requirements.txt	update requirements.txt in /docs (#101092 )	2023-05-12 03:19:36 +00:00

Please see the Writing documentation section of CONTRIBUTING.md for details on both writing and building the docs.