strisunshinewentingwang/pytorch

Pull Requests Commits

Enable softmax and tiny norm FP16 tests on ROCm (#46363)

ashish committed 3 years ago

88e94da5

Add ReplaceNaN benchmark as baseline (#46685)

ShijunK committed 3 years ago

6ae0a7c9

Make add_relu an internal function (#46676)

albanD committed 3 years ago

27e2ea4c

Enable DataParallel to run zero input Module (#46565)

lixinyu committed 3 years ago

870a5a0d

[quant][fx] EmbeddingBag quantization support (#46678)

supriyar committed 3 years ago

842494af

[quant][fx] Embedding quantization support (#46677)

supriyar committed 3 years ago

e34c825b

Clean up use of Flake8 in GitHub CI (#46740)

samestep committed 3 years ago

fe6fb775

[CI][IOS] Add a arm64 ios job for Metal (#46646)

xta0 committed 3 years ago

bf1ea14f

[CI][IOS] Rename the IOS_VERSION (#46645)

xta0 committed 3 years ago

344abd56

ProcessGroupNCCL::alltoall_base needs to call recordStream (#46603)

jeffdaily committed 3 years ago

ce5bca55

[quant][graphmode][fx] Add support for additional_fuse_method_mapping (#46345)

jerryzh168 committed 3 years ago

bd90379d

[pt][static_runtime] Add option enable_out_variant (#46690)

Hao Lu committed 3 years ago

d6519d4e

Remove dilation restriction on cuDNN ConvTranspose2d (#46290)

xwang233 committed 3 years ago

f326f6a8

[caffe2] Fix inplace ops in onnx::SsaRewrite (#46134)

Hao Lu committed 3 years ago

53dff784

[caffe2] Allow memonger to optimize nets with inplace(enforced) ops (#46560)

Hao Lu committed 3 years ago

51bf7bed

[quant][graphmode][fx] Add additional_qat_module_mapping (#46344)

jerryzh168 committed 3 years ago

23fad911

torch.nn.Unfold accepts 0-dim for batch size (#40689)

v0dro committed 3 years ago

982fa07c

Revert "Push rocm to slow path (#46216)" (#46728)

Iurii Zdebskyi committed 3 years ago

c57c5607

[FX] Make wrapped functions traceable (#46692)

James Reed committed 3 years ago

9ccf85b7

[FX] Fix recursion depth issue on Graph deepcopy (#46669)

James Reed committed 3 years ago

2700932e

Batching rules for: new_zeros, new_empty (#46606)

zou3519 committed 3 years ago

18d80501

Clarify timing of GetDeviceProperty() (#46715)

r-barnes committed 3 years ago

c4430088

[OpBench] fix jit mode run of operator benchmark for ops with parameters (#46694)

Yang Wang committed 3 years ago

920ec665

Pull in fairscale.nn.Pipe into PyTorch. (#44090)

pritamdamania committed 3 years ago

06d50b5e

[OSS][Metal] Support Resnet models

xta0 committed 3 years ago

b63ddd6f

Replace map(lambda constructs (#46462)

Flamefire committed 3 years ago

93719440

[RPC] print exception message on workers that run python functions (#46372)

rohan-varma committed 3 years ago

25dc0056

[py][vulkan][reland] Add is_vulkan to py api, add vulkan to device type parsing (#46655)

IvanKobzarev committed 3 years ago

3112e234

Push rocm to slow path (#46216)

Iurii Zdebskyi committed 3 years ago

bc1ce584

Add comment about running C++ executable lint locally (#46698)

samestep committed 3 years ago

3526b604

Newer Older