fduwjj 0003d5135d [TP] Enable partial tensor add without redistribute (#105939) ... Pull Request resolved: https://github.com/pytorch/pytorch/pull/105939 Approved by: https://github.com/wanchaol		2023-07-26 03:12:39 +00:00
..
__init__.py	[DTensor][1/N] add DTensor RNG state APIs (#98198)	2023-04-10 23:57:00 +00:00
basic_strategy.py	[dtensor] add StrategyType and TupleStrategy (#99435)	2023-04-22 05:39:20 +00:00
common_rules.py	[dtensor] tensor ops to use strategy based sharding prop (#100607)	2023-05-11 02:47:20 +00:00
embedding_ops.py	[PTD][TP] Add BWD support for colwise embedding sharding (#104820)	2023-07-10 22:33:20 +00:00
math_ops.py	Add missing torch.distributed.ReduceOp.AVG in type stubs (#101534)	2023-05-16 15:51:21 +00:00
matrix_ops.py	[dtensor][6/N] change to a better/safer op registration (#90735)	2023-02-01 05:06:33 +00:00
pointwise_ops.py	[TP] Enable partial tensor add without redistribute (#105939)	2023-07-26 03:12:39 +00:00
random_ops.py	[DTensor][3/N] enable aten.native_dropout (#98577)	2023-04-10 23:57:04 +00:00
tensor_ops.py	[TP] Use Stride inferred from local tensor in to_local bwd (#102630)	2023-06-01 04:30:24 +00:00
utils.py	[Reland] Update mypy to 1.4.1 (#105227)	2023-07-15 20:30:20 +00:00
view_ops.py	[spmd] quick fix on batch input view issue (#98813)	2023-04-11 14:27:01 +00:00