strisunshinewentingwang/pytorch

Pull Requests Commits

fix numpy compatibility in test for `torch.kthvalue` (#59214)

pmeier committed 3 years ago

0a267819

Fix device of info tensor for torch.linalg.inv_ex with MAGMA backend (#59223)

IvanYashchuk committed 3 years ago

e9e1bb1a

[quant][graphmode][fx][refactor] Remove Quantizer class from quantize_node (#59039)

jerryzh168 committed 3 years ago

50e6ee3c

CUDA support in the CSR layout: CUDA addmm/matvec (#59012)

aocsa committed 3 years ago

2d8f0d96

[CUDA graphs] Makes sure all graphs tests call empty_cache() at some point before capture (#59233)

mcarilli committed 3 years ago

3efefc40

[quant][graphmode][fx][refactor] Remove _prepare from Quantizer class (#59038)

jerryzh168 committed 3 years ago

1d37f415

[Reland] Adds an aten::_ops namespace with unambiguous function names (#59018)

zou3519 committed 3 years ago

970096b6

use long index type for index_add_cuda deterministic path (#59254)

yuguo68 committed 3 years ago

8805093e

[quant][graphmode][fx][refactor] Remove find_matches from Quantizer class (#59037)

jerryzh168 committed 3 years ago

20348fb3

[quant][graphmode][fx][refactor] Remove fold_weights from Quantizer class (#59036)

jerryzh168 committed 3 years ago

7d64fc67

DOC Adds register_module_full_backward_hook into docs (#58954)

thomasjpfan committed 3 years ago

8af62812

[nnc] Enable CPU fusion inside Facebook, take 3 (#59253)

bertmaher committed 3 years ago

6e7dae9c

[quant][graphmode][fx][refactor] Remove save_state and restore_state from Quantizer class (#59035)

jerryzh168 committed 3 years ago

cc489180

Fix mismatch in README.md Docker Image section (#59199)

ChenYuHo committed 3 years ago

336ac949

[ROCm] disable test test_Conv2d_groups_nobias for ROCm (#59158)

jaglinux committed 3 years ago

95c26b28

[quant][graphmode][fx][refactor] Remove prepare_custom_config from Quantizer class (#59034)

jerryzh168 committed 3 years ago

3d521e8b

Revert D28240105: [pytorch][PR] Fix DistributedSampler mem usage on large datasets

rohan-varma committed 3 years ago

a5dcd3c4

Fix DistributedSampler mem usage on large datasets (#51841)

amccollum committed 3 years ago

a0ce8da2

Add NCCL_ASYNC_ERROR_HANDLING as an environment variable (#59109)

awgu committed 3 years ago

5a42a97c

DOC Update register_buffer/parameter docstring explaining None (#59015)

thomasjpfan committed 3 years ago

5f111722

[quant][graphmode][fx][refactor] Remove patterns from Quantizer class (#59033)

jerryzh168 committed 3 years ago

e4b26843

[quant][graphmode][fx][refactor] Remove node_name_to_scope from Quantizer (#59032)

jerryzh168 committed 3 years ago

83892c18

[quant][graphmode][fx][refactor] Remove quantized_graph from Quantizer (#59031)

jerryzh168 committed 3 years ago

3826f7e8

[quant][gx][graphmode][refactor] Remove modules from Quantizer (#59030)

jerryzh168 committed 3 years ago

1b4586ee

Add check_env, getenv api (#59052)

1ntEgr8 committed 3 years ago

aa857850

Fixed torch.nn.MultiMarginLoss equation format error (#59188)

aflah02 committed 3 years ago

fd2a3636

Create EngineHolder for serializing and running TRT Engines with PyTorch

jackm321 committed 3 years ago

06399d44

Improve Tensor traverse to traverse its grad_fn when possible (#58271)

albanD committed 3 years ago

e9e55885

Un-verbose the build (#59235)

driazati committed 3 years ago

65748f81

[quant][graphmode][fx] Factor out run_weight_observer (#59029)

jerryzh168 committed 3 years ago

75237283

Newer Older