strisunshinewentingwang/pytorch

Pull Requests Commits

[wip][quant][graphmode] produce reference pattern for binary ops and then rewrite to quantized op (#68229)

jerryzh168 committed 2 years ago

3e7b9494

[fx2trt][easy] add sparse flag to TRTInterpreter (#69495)

842974287 committed 2 years ago

fd6ad018

[fx2trt] Separate sign from `trunc_div` and use it for acc_ops.sign (#69486)

842974287 committed 2 years ago

155be3cd

[fx2trt] [Prep for release] Add type hints to converters and separate main files (#69458)

842974287 committed 2 years ago

df20d9e2

[fx2trt] Add a helper function to generate specs for dynamic batch size (#69405)

842974287 committed 2 years ago

d7d6260b

[fx2trt] Use weights shape as normalize shape in layer norm (#69401)

842974287 committed 2 years ago

7c7cb94a

Revert D32704467: [Autograd/Checkpoint] Checkpoint implementation without reentrant autograd

suo committed 2 years ago

fa3bf8b3

Initialize upgrader and operator version files (#68772)

tugsbayasgalan committed 2 years ago

42933e1e

[Pytorch Edge] Add new_empty_strided to tracer (#69492)

JacobSzwejbka committed 2 years ago

c1b81b48

[PyTorch][Distributed] Use auto-grad enabled collections for the shared linear op to enable backward grad calculation (#68096)

fduwjj committed 2 years ago

1f06ac37

[PyTorch][Distributed] Enable Reduce Scatter and modify all_to_all for sharded linear with more test cases. (#68786)

fduwjj committed 2 years ago

de695f3d

[Autograd/Checkpoint] Checkpoint implementation without reentrant autograd (#69027)

rohan-varma committed 2 years ago

1574ce1a

add VSX dispatch for fft_fill_with_conjugate_symmetry_stub (#68914)

cdeepali committed 2 years ago

f6b1d5d2

.github: Volume mount local netrc for docs push (#69472)

seemethere committed 2 years ago

0a10aa64

[FSDP] Enforce wrapper_cls as a mandatory kwarg in enable_wrap. (#69358)

rohan-varma committed 2 years ago

2781b198

[FSDP] Kill config_auto_wrap_policy, remove policy from enable_wrap, (#69357)

rohan-varma committed 2 years ago

95d26257

[FSDP] Remove auto_wrap() (#69356)

rohan-varma committed 2 years ago

e2a9ea1b

[FSDP] Make recursive_wrap, wrap APIs independent of ConfigAutoWrap. (#68776)

rohan-varma committed 2 years ago

02600ced

FX pass: fuse_sparse_matmul_add (#69340)

Marat Subkhankulov committed 2 years ago

76749b61

[Vulkan] Implement slice operator (#69382)

beback4u committed 2 years ago

2183ad69

unify compare kernels (#69111)

ngimel committed 2 years ago

2c2e7f63

Clean up CODEOWNERS for .github/ (#69395)

seemethere committed 2 years ago

28984032

Move THC_sleep to ATen (#69038)

peterbell10 committed 2 years ago

6eb1cddd

Skips failing ROCm test (#69456)

Mike Ruberry committed 2 years ago

1ddaed64

expose return_types in Python (#66614)

kshitij12345 committed 2 years ago

85bbe597

The Jiterator (#69439)

Mike Ruberry committed 2 years ago

e2fec90e

Remove THGeneral (#69041)

peterbell10 committed 2 years ago

b08d6420

[ROCm] Refactor how to specify AMD gpu targets using PYTORCH_ROCM_ARCH (#61706)

jithunnair-amd committed 2 years ago

8dfdc3df

[SR][easy] Accessors for value array offsets (#69755)

Mike Iovine committed 2 years ago

c6c3b434

Add lazy impl of normal_ (#69743)

wconstab committed 2 years ago

Verified a76899b0

Newer Older