strisunshinewentingwang/pytorch

Pull Requests Commits

add rsub decomposition with alpha (#94144)

Natalia Gimelshein committed 1 year ago

7bba87ed

trymerge to ignore certain failures (#91134)

clee2000 committed 1 year ago

e9533767

COO intersection kernel: respect value intersection order (#92242)

nikitaved committed 1 year ago

b07c839b

[Py-3.11] Skip dynamo related tests (#94187)

albanD committed 1 year ago

0b2dc3b3

[MPS] Skip gather/blit calls in case of strided output (#94260)

DenisVieriu97 committed 1 year ago

5d48392a

[MPS] Fix MPSGraph casting issue to MPSDataTypeBool in masked_fill op (#94263)

DenisVieriu97 committed 1 year ago

86ae14de

[MPS] Fallback on gather code to solve view tensors when a slice is followed by a reshape (#94278)

DenisVieriu97 committed 1 year ago

e3ac1096

[MPS] Raise error for int64 inputs of dot operator. (#94270)

kulinseth committed 1 year ago

4cd086b1

[MPS] Fix the argument error for tensor_split() test (#94234)

razarmehr committed 1 year ago

b654d149

[MPS] Remove the unused code for view lists in OperationUtils.h (#94265)

razarmehr committed 1 year ago

a3ca66c6

[MPS] Don't reset the Graph state (#94283)

kulinseth committed 1 year ago

a0a37280

[MPS] Fix the crash in View ops when slicing wrong lengths (#94259)

razarmehr committed 1 year ago

36062dd2

General in-place binary op support in dynamo (#94203)

jbschlosser committed 1 year ago

bf4fe5dd

Dynamo: Fix to unpack ConstantVariable in call_range() (#94202)

jbschlosser committed 1 year ago

f954498e

Fix thread-allocation in `_vec_log_softmax_lastdim` (#85398)

sanchitintel committed 1 year ago

c4544bc1

update test fixture (#89796)

eellison committed 1 year ago

a2ac25f6

sparse compressed tensor validation without syncs for low-(batch)dim tensors. (#94048)

nikitaved committed 1 year ago

513b5da3

[BE] Add empty tensor check to _compute_linear_combination (#94245)

malfet committed 1 year ago

42b6bcdb

[Inductor] Fix CPU vectorized implementation of mask calculation that breaks torch.where (#93922)

jgong5 committed 1 year ago

a28a0629

[inductor] bug fix: use `create_symbolic_sizes_strides_storage_offset` (#94031)

nkaretnikov committed 1 year ago

0e94fbc0

[Dynamo] Support torch.Tensor.fn as TorchVariable, not UserDefinedObjectVariable, preventing graph break (#93243)

min-jean-cho committed 1 year ago

900e09c8

Refactor sharding data pipe into a seperate file (#94095)

wenleix committed 1 year ago

d6dec1a5

[quant][fx][pt2e] Refactor prepare so it's aligned better with the new API plan in pt2e (#94011)

jerryzh168 committed 1 year ago

59c1b502

[Docs] Add pointer to FlashAttention paper (#94253)

tridao committed 1 year ago

ffb3561c

Clean up mentions of removed torch/csrc/generic/*.cpp (#94107)

kit1980 committed 1 year ago

f92348e1

[MPS] Unregister put_() op due to lack of implementation (#94231)

razarmehr committed 1 year ago

bc8a3783

[FSDP][optim_state_dict] Let optim_state_dict ignore the non-FSDP managed parameters that do not reside on the rank (#94129)

fegin committed 1 year ago

bc6d54f6

[FSDP][state_dict] Fix incorrect valid_data_size for local_state_dict when some ranks have zero data. (#94109)

fegin committed 1 year ago

f04106f1

FakeTensor should constant propagate through ops that allow numbers as scalars (#94145)

yanboliang committed 1 year ago

605b6618

[mobile] List all missing ops at once (#94205)

larryliu0820 committed 1 year ago

579ae64d

Newer Older