strisunshinewentingwang/pytorch

Pull Requests Commits

more passing dynamo tests (#121378)

aorenste committed 251 days ago

23ac0cd5

[nit][DCP][DSD] Remove Unused Variables in test_state_dict.py (#121204)

wz337 committed 251 days ago

4186c365

Revert "[fake_impls] Fix seed/offset device for attention kernels (#120839)" (#121447)

davidberard98 committed 251 days ago

0f8c9acc

[dtensor][TP] check funcol calls and improve doc for loss parallel (#121366)

tianyu-l committed 251 days ago

dc514b96

[fx] Preserve Fx graph node order in partitioner across runs (#115621)

kareemshaik80 committed 251 days ago

25c74a93

make dyanmo work with _LazyGraphModule.lazy_forward (#121259)

shunting314 committed 251 days ago

7dc1ab89

[Torch Elastic][Draft] Refactor SubprocessHandler to separate module for easier subclass (#120373)

cniii committed 251 days ago

9bff1599

[dynamo][guards-cpp-refactor] Func defaults and kwdefaults accessor (#121338)

anijain2305 committed 251 days ago

c86a1ce1

[dynamo][guards-cpp-refactor] Permit dict version guard in DictGuardManager (#121327)

anijain2305 committed 251 days ago

79a04f2d

Update XNNPACK revision to fcbf55a (#120583)

GregoryComer committed 251 days ago

962c1b4c

[Indutor] Support auto-tuned custom PT ops in abi compatible mode (#120877)

ColinPeppler committed 251 days ago

090616d9

[dynamo][guards] Use lazy variable tracker for func defaults (#121388)

anijain2305 committed 251 days ago

04a5d6e8

Fixup test_trace_rules (#121351)

izaitsevfb committed 251 days ago

5d8e4126

[export] Fix nn_module_stack in retracing (#121423)

angelayi committed 251 days ago

af62a70f

Clean up mode handling in python dispatcher (#121083)

tugsbayasgalan committed 251 days ago

4f120dc2

[DCP][state_dict] Let _offload_state_dict_to_cpu to return the companion_obj if it exist. (#121273)

fegin committed 251 days ago

0811f152

[BE] NO MORE discrepancy between forloop foreach capturable YAY (#121269)

janeyx99 committed 251 days ago

f76e541e

Add ASGD capturable API for forloop (#121264)

janeyx99 committed 251 days ago

9d6c5be7

Add RAdam capturable API for forloop (#121260)

janeyx99 committed 251 days ago

24821fec

feat: Add min, max ranges to mark_dynamic API (#119737)

peri044 committed 251 days ago

b1657bea

Revert "[Inductor] Add support for NEON ISA in the Inductor C++ backend (#105590)"

pytorchmergebot committed 251 days ago

e0c534fe

Add torch.cond support to AOT Inductor (#121120)

aakhundov committed 251 days ago

3d089de8

Remove unnecessary use of ctx.resolve_tools. (#120493)

tjgq committed 251 days ago

26740f85

[dynamo] add more refleak tests (#120657)

williamwen42 committed 251 days ago

d14d62b7

Remove dead get_shape_groups (#120813)

ezyang committed 251 days ago

6490441d

[Inductor] Use indices for constants in triton_meta (#121427)

oulgen committed 251 days ago

18d574a0

Fix for Wait kernel lowering in inductor not accepting MultiOutputs from non-collective calls (#121428)

ani300 committed 251 days ago

f61192b0

[export] Serialize union fields with single entry dict. (#121263) (#121337)

zhxchen17 committed 251 days ago

76f14618

[PyTorch] Use uint32_t for ProcessedNode::num_outputs (#121335)

swolchok committed 251 days ago

4c58f2b6

Subclass view fake-ification via reified ViewFuncs (#118405)

jbschlosser committed 251 days ago

ea8f6e2e

Newer Older