strisunshinewentingwang/pytorch

Pull Requests Commits

Remove overly restrictive assert in adam (#80222)

albanD committed 2 years ago

375668cd

[quant] Quantizable documentation (#79957)

HDCharles committed 2 years ago

0308609b

[ao][sparsity] Implemented state_dict() and load_state_dict() functions (#79883)

macandro96 committed 2 years ago

524d1812

[quant] Add QuantizedMHA class (#79956)

HDCharles committed 2 years ago

3bcec850

[ao][sparsity] Implemented the step() function (#79822)

macandro96 committed 2 years ago

af4e2b2c

[ao][sparsity] Base scheduler class for Data Schedulers (#79817)

macandro96 committed 2 years ago

70b7bca4

[primTorch] asinh, atanh (#80210)

khushi-411 committed 2 years ago

b00448df

Cleanup buck_setup.sh a bit (#80198)

malfet committed 2 years ago

2d7d5a75

Revert "[FakeTensor] Use the device of the meta tensor for fallback kernel (#80193)"

pytorchmergebot committed 2 years ago

35268bdc

[ao] Adds framework for InputWeightEqualization Detector (#79916)

vspenubarthi committed 2 years ago

845021db

Support softmax(nested tensor) (#80179)

YifanShenSZ committed 2 years ago

54a1cc52

Revert "Raise warning for unpickable local function (#80140)"

pytorchmergebot committed 2 years ago

79ba65c0

Add forward AD for linalg.det and simplify its backward (#79487)

lezcano committed 2 years ago

42a23596

Raise warning for unpickable local function (#80140)

ejguan committed 2 years ago

4b75b7d3

[Static Runtime] Add Metadata to ProcessedNode depending upon the op type (#79961)

akshayParashar1995 committed 2 years ago

3afc802c

[Vulkan] Optimize LSTM operator with pre-packing (#79702)

manuelcandales committed 2 years ago

bab1ea85

Setting validation flag for Distributions tests to work with TorchDynamo (#80081)

anijain2305 committed 2 years ago

6f9460bd

[GHA] Add warning when S3 stats for sharding aren't found (#80176)

janeyx99 committed 2 years ago

addeb1ed

[FakeTensor] Use the device of the meta tensor for fallback kernel (#80193)

eellison committed 2 years ago

93e70c59

[torchdynamo hash update] update the pinned torchdynamo hash (#80112)

pytorchmergebot committed 2 years ago

18b6ce38

[xla hash update] update the pinned xla hash (#80196)

pytorchmergebot committed 2 years ago

219692d2

sync pt_ops.bzl (#80194)

linbinyu committed 2 years ago

55aa0b1a

Port cholesky to structured kernels (#79300)

lezcano committed 2 years ago

ff5a588e

switch computation to computationPtr (#79491)

Krovatkin committed 2 years ago

7bf1b1dc

Make linalg_det function structured (#79486)

lezcano committed 2 years ago

ea5299fc

Add support for indexing cuda tensors with cpu (#80115)

eellison committed 2 years ago

7e54959a

Use streams for import_ir_module for pickle case (#80131)

qihqi committed 2 years ago

648224dd

BUG: Evade segfault by throwing a RuntimeError for `nn.ChannelShuffle` and empty input tensors (#77029)

HaoZeke committed 2 years ago

72e40d2b

[ci] delete JOB_BASE_NAME (#80046)

suo committed 2 years ago

5029a91f

Use cub::BlockRadixSort to improve medium length sort performance (#79628)

peterbell10 committed 2 years ago

8c0796e5

Newer Older