strisunshinewentingwang/pytorch

Pull Requests Commits

[quant][fix] Fix quant type classification for float_qparam qconfig (#48069)

jerryzh168 committed 3 years ago

576fa091

Introducing winograd transformed fp16 nnpack to PT for unet 106 (#47925)

kimishpatel committed 3 years ago

f0f8b97d

[PyTorch] Make RecordFunction::active private (#47549)

swolchok committed 3 years ago

383abf1f

[PyTorch][JIT] Skip unnecessary refcounting in TensorType::merge (#47959)

swolchok committed 3 years ago

1bafff23

Removing non-thread-safe log statement from ReinitializeTensor (#48185)

Max Ouellet committed 3 years ago

0f89be61

pass strict_fuser_check for recursive fusion (#47221)

jjsjann123 committed 3 years ago

43604863

Revert D24853669: [pytorch][PR] Migrate `eig` from the TH to Aten (CUDA)

Mike Ruberry committed 3 years ago

ea1e78a0

fft: Generalize fill with conjugate symmetry and use complex dtypes (#46908)

peterbell10 committed 3 years ago

2fbd70d3

move Tensor comparisons back to C (#48018)

Taylor Robie committed 3 years ago

0639387f

move aten::round to lite interpreter (#45931)

linbinyu committed 3 years ago

ed4dd865

[pytorch][codegen] simplify python signature creation logic (#47977)

ljk53 committed 3 years ago

a36e6468

[pytorch][codegen] simplify dunder method check in gen_python_functions.py (#47976)

ljk53 committed 3 years ago

5eaf8562

[pytorch][codegen] remove dead code in gen_variable_type.py (#47975)

ljk53 committed 3 years ago

52434567

[tensorexpr] Switch cpp tests to pure gtest (#48160)

bertmaher committed 3 years ago

07657b60

[te][benchmark] Add more optimized versions of gemm (#48159)

bertmaher committed 3 years ago

464d23e6

[te] Make BUILD_TENSOREXPR_BENCHMARK a real CMake option (#48158)

bertmaher committed 3 years ago

8a996dd1

Migrate `eig` from the TH to Aten (CUDA) (#44105)

antocuni committed 3 years ago

866f8591

Revert D24924736: [pytorch][PR] Hipify revamp

malfet committed 3 years ago

8af9f2cc

Add `torch.swapdims` and `torch.swapaxes` (#46041)

kshitij12345 committed 3 years ago

68a3a3f3

[JIT] Pass TypePtr by reference in Argument::type() and Type::isSubtypeOfExt(). (#48061)

Mikhail Zolotukhin committed 3 years ago

d256e388

Document that `remainder` does not support complex inputs (#48024)

kurtamohler committed 3 years ago

df88cc3f

Fix default value of `num_replicas` in DistributedSampler docstring (#48135)

mx-iao committed 3 years ago

0387f2a6

Disable distributed collectives profiling tests (#48129)

rohan-varma committed 3 years ago

140e946f

Small documentation changes for RRef and Dist Autograd (#48123)

H-Huang committed 3 years ago

a6898cb5

Enable complex tests that depend on batched matmul on CUDA (#47910)

IvanYashchuk committed 3 years ago

81b1673a

Install magma on CUDA 11.1 (#48164)

zasdfgbnm committed 3 years ago

3ca4c656

Hipify revamp (#45451)

jithunnair-amd committed 3 years ago

10b490a3

Make numpy optional dependency for torch.cuda.amp (#48154)

malfet committed 3 years ago

1454cbf0

Enable the faster combined weight branch in MHA when query/key/value is same object with nan (#48126)

Guanheng Zhang committed 3 years ago

e2b4c63d

Add max supported SM for nvrtc-11.0 (#48151)

malfet committed 3 years ago

9ead5588

Newer Older