strisunshinewentingwang/pytorch

Pull Requests Commits

[Doc] Add weight dtype in torch.nn.CrossEntropyLoss (#110998)

isdanni committed 344 days ago

6c7013a3

[quant][pt2e] Disable remove_qconfig (#111000)

jerryzh168 committed 344 days ago

d589106b

enable index add test (#111016)

eellison committed 344 days ago

cf1da9bd

Clean-up composite implicit ops for aten::isfinite, isreal and log_sigmoid (#110896)

guilhermeleobas committed 344 days ago

e151307d

Revert "[2/N] Dynamo supports skip by function & removes skipfiles circular import (#110835)"

pytorchmergebot committed 344 days ago

d3205f83

[2D] Enable 2D FSDP+TP model.load_state_dict() (#110925)

wz337 committed 344 days ago

80dfc974

Implement tensor slice in inductor to stop falling back for aten.index (#111015)

oulgen committed 344 days ago

fd4ba806

[2D] Enable 2D DTensor state_dict for FSDP + TP (#110846)

wz337 committed 344 days ago

6c136c33

[2/N] Dynamo supports skip by function & removes skipfiles circular import (#110835)

yanboliang committed 344 days ago

0bd4ce72

[dtensor] small change to refactor random ops (#110900)

wanchaol committed 344 days ago

de1ca4a0

[dtensor] make replicate -> partial do division instead (#110898)

wanchaol committed 344 days ago

657e8f2c

Reland [Profiler] Improve the docstring for export_memory_timeline (#110983)

aaronenyeshi committed 344 days ago

204f338f

Revert "[sparse] Add i8i8->i32 support for cuSPARSELt (#110499)"

pytorchmergebot committed 344 days ago

cae3a2e6

[aotinductor] Add both cpu and cuda tests for the AOTInductor cpp test (#110920)

desertfire committed 344 days ago

86619c9c

[aotinductor] Add AOTIModelRunner as a utility class (#110891)

desertfire committed 344 days ago

3058700f

[aotindutor] Update the cpp test example (#110652)

desertfire committed 344 days ago

b17c247e

[cond] Add more tests for valid inputs of cond (#110727)

ydwu4 committed 344 days ago

3062e267

Suppress warnings in tensorpipe.h (#111012)

malfet committed 344 days ago

ef19824d

[MPS][BE] Avoid redispatch in `sign_out` (#110955)

malfet committed 344 days ago

f2d47684

[inductor] Implement Fx graph caching to improve warm compilation time. (#103453)

masnesral committed 344 days ago

fc1105b2

[generate_opcheck_tests] Add some reasonable defaults (#110977)

zou3519 committed 344 days ago

2cf97829

[aotinductor] Add a perf smoke test for AOTInductor (#110972)

desertfire committed 344 days ago

4abfa228

Revert "[core ATen IR] Add decompositions for max, min, var_mean (#110906)"

pytorchmergebot committed 344 days ago

98c329b1

[MPS] Add support for Softshrink to MPS Backend (#110814)

igm503 committed 345 days ago

95ff51d8

[Distributed] Small nits to apply_optimizer_in_backward (#110903)

rohan-varma committed 345 days ago

de370eb3

Revert "[export] Get export APIs ready for PTC (#110410)"

pytorchmergebot committed 345 days ago

08218681

Fix broken period workflow after #110976 (#111013)

huydhn committed 345 days ago

74029fae

[MPS] Use Metal Events to synchronize buffers in MPSAllocator (Part 1) (#106938)

razarmehr committed 345 days ago

056d6247

[export] Get export APIs ready for PTC (#110410)

angelayi committed 345 days ago

b96ea9f3

Revert "Reland 3rd try [finishing colesbury's PR 100642] Guard on nn.Module dicts and type (#109323)" + Forward fixes + test (#110964)

voznesenskym committed 345 days ago

1e7947b3

Newer Older