strisunshinewentingwang/pytorch

Pull Requests Commits

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

ColinPeppler committed 196 days ago

090616d9

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

anijain2305 committed 196 days ago

04a5d6e8

Fixup test_trace_rules (#121351)

izaitsevfb committed 196 days ago

5d8e4126

[export] Fix nn_module_stack in retracing (#121423)

angelayi committed 196 days ago

af62a70f

Clean up mode handling in python dispatcher (#121083)

tugsbayasgalan committed 196 days ago

4f120dc2

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

fegin committed 196 days ago

0811f152

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

janeyx99 committed 196 days ago

f76e541e

Add ASGD capturable API for forloop (#121264)

janeyx99 committed 196 days ago

9d6c5be7

Add RAdam capturable API for forloop (#121260)

janeyx99 committed 196 days ago

24821fec

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

peri044 committed 196 days ago

b1657bea

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

pytorchmergebot committed 196 days ago

e0c534fe

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

aakhundov committed 196 days ago

3d089de8

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

tjgq committed 196 days ago

26740f85

[dynamo] add more refleak tests (#120657)

williamwen42 committed 196 days ago

d14d62b7

Remove dead get_shape_groups (#120813)

ezyang committed 196 days ago

6490441d

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

oulgen committed 196 days ago

18d574a0

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

ani300 committed 196 days ago

f61192b0

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

zhxchen17 committed 196 days ago

76f14618

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

swolchok committed 196 days ago

4c58f2b6

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

jbschlosser committed 196 days ago

ea8f6e2e

TD Heuristic for tests mentioned in PR body, less verbose TD printing (#120621)

clee2000 committed 196 days ago

63ec5cd1

[CI] Script to fetch creds from current AWS session (#121426)

malfet committed 196 days ago

c7a65f58

Revert "[compiled autograd] support custom ops backed by c++ autograd::Function (#120681)"

pytorchmergebot committed 196 days ago

2b1661c7

create function to get ProcessGroupNCCL uid (#121132)

shengbao-zheng committed 196 days ago

60aaba41

[BE] Remove unnecessary requires_cuda in common_optimizers.py (#121249)

janeyx99 committed 196 days ago

83d095c2

Add capturable single tensor Adamax (#121183)

janeyx99 committed 196 days ago

53bdae73

Forward fix lint after 121202 (#121425)

clee2000 committed 196 days ago

af88425c

[export] update docs to not export raw functions (#121272)

suo committed 196 days ago

c3c15eb9

Revert "[ATen][CUDA][CUBLAS] cublasLtMatmul increase workspace_size (#120925)"

pytorchmergebot committed 196 days ago

862b99b5

[profiler] record nccl version in distributed info (#121044)

shengbao-zheng committed 196 days ago

eea37c6d

Newer Older