strisunshinewentingwang/pytorch

Pull Requests Commits

[PyTorch] Round T up to next multiple of 8 in NestedTensor case

swolchok committed 2 years ago

7e4730d0

Use random seed in normal_mps_out (#78010)

lhoenig committed 2 years ago

51c4c79e

Add MPS merge rules (#78259)

malfet committed 2 years ago

fc2aef55

Build Python module and tests CPU mode with Bazel in CI

dreiss committed 2 years ago

8449ac77

[PT-D] Enable nan_to_num op for sharded tensor

fduwjj committed 2 years ago

141238a8

[quant][core][better-engineering] Rename files in quantized/cuda directory to conform with non-quantized countertpart filenames #77431

dzdang committed 2 years ago

a9b68009

Add lazy shape inference for logical boolean ops (#78004)

wonjoolee95 committed 2 years ago

593d66e1

torch/distributed: move WorkerInfo registration into libtorch instead of libtorch_python (#78028)

d4l3k committed 2 years ago

ebba4219

[FSDP] Remove unneeded padding logic for optim state dict

awgu committed 2 years ago

8412f209

Add from_blob with storage_offset arg

JacobSzwejbka committed 2 years ago

cdb009c8

ns for fx: remove restriction on nodes with no args and only kwargs

vkuzo committed 2 years ago

53e05ad4

Fix typo in testname (#78258)

malfet committed 2 years ago

70508265

[Profiler] Abstract interface for Python tracer

Taylor Robie committed 2 years ago

e0a071a4

[Profiler] Build call tree in `collection.cpp`

Taylor Robie committed 2 years ago

34d160b1

[XLA hash update] update the pinned xla hash (#78144)

pytorchmergebot committed 2 years ago

9f9e8ff2

ci: Remove verbose windows compilation commands (#78197)

clee2000 committed 2 years ago

7e767729

Conversion between SparseBsr and Strided (#78025)

cpuhrsch committed 2 years ago

b9fb940d

[MPS] Lazy initialize allocators (#78227)

malfet committed 2 years ago

2679aa47

[quant][core][improvement][bug fix] Added channel axis bound checking in fused_moving_avg_obs_fake_quant_*

dzdang committed 2 years ago

ba952d83

[quant][core][gpu][improvement] Made plan and run for quantized linear op conform with Conv_v8.cpp

dzdang committed 2 years ago

a98b1a8b

[quant][core][gpu][improvement] Removed linear_output and set output tensors as virtual in quantized cudnn linear op

dzdang committed 2 years ago

b4471148

Pass WITH_BLAS option from environment to CMake (#78037)

Flamefire committed 2 years ago

d5f99581

[quant][core][gpu][improvement] Made plan and run for quantized cudnn conv op conform with Conv_v8.cpp

dzdang committed 2 years ago

589f40e3

Revert "[quant] Add utility function get_fqn_to_example_inputs"

pytorchmergebot committed 2 years ago

87148f2b

[PyTorch] Fix record function inputs_valid_ check (#78002)

louisfeng committed 2 years ago

82cb7210

remove torch.no_grad from sample_inputs (#78076)

kshitij12345 committed 2 years ago

17c1aed2

[quant][core][gpu][improvement] Removed conv_output and set output tensors as virtual in quantized cudnn conv2d op

dzdang committed 2 years ago

bd5ec6c8

[quant] Add utility function get_fqn_to_example_inputs

jerryzh168 committed 2 years ago

50a44fe4

[quant][core][gpu][improvement] Enabled broadcasting multiplication support for requantize_multiplier_tensor in quantized cudnn add, linear, and conv2d ops

dzdang committed 2 years ago

b2917526

ezyang committed 2 years ago

a1765f01

Newer Older