strisunshinewentingwang/pytorch

Pull Requests Commits

Throw a nice error when SubTensor.__torch_dispatch__() returns the wrong type for detach()

jbschlosser committed 2 years ago

0794d59d

Support no-batch-dim for CrossEntropyLoss with prob target

jbschlosser committed 2 years ago

8881d7ac

rocblas alt impl during backward pass only (#71881)

jeffdaily committed 2 years ago

de86146c

Revert "Adds torch.cuda.is_current_stream_capturing (#77673)"

pytorchmergebot committed 2 years ago

0d8a0f18

Upstream master bump 0513 (#77471)

jjsjann123 committed 2 years ago

a2802ad0

Revert "Revert "Implement sym_sizes to create proper IR for sym ints representing tensor sizes (#76836)""

ezyang committed 2 years ago

4941e72e

ezyang committed 2 years ago

befa4e37

Fix 'Code below assumes there is at least one tensor arg' assumption (#76917)

antoniojkim committed 2 years ago

55be35ae

Add sharding tests to multigpu-test.sh (#77708)

pritamdamania87 committed 2 years ago

a7cf95a6

Adding a way to register both upper and lower bound functions

Gamrix committed 2 years ago

2a990181

Adding support for nonzero in LazyTensor shape functions

Gamrix committed 2 years ago

73480bcb

[fix] composite compliance : nuclear_norm (#77734)

kshitij12345 committed 2 years ago

23b2e882

[nvFuser] Improving bitwise ops support (#77158)

zasdfgbnm committed 2 years ago

4eec865f

[chalf] div, eq, masked_fill, index_put (#77479)

kshitij12345 committed 2 years ago

85710070

Adds torch.cuda.is_current_stream_capturing (#77673)

mcarilli committed 2 years ago

d03d43df

Micro-optimisations for matmul 2.0: Electric boogaloo

lezcano committed 2 years ago

ddb2eb7a

Use any_type in test_out

lezcano committed 2 years ago

c446f78f

[DataPipe] Update `mux` data pipe (#76384) (#77145)

ninginthecloud committed 2 years ago

4d1ead6d

[FSDP] Warning for shared params, small doc fixes (#77726)

rohan-varma committed 2 years ago

4c343432

Add option to disable autocast pass

Elias Ellison committed 2 years ago

05ce0f9b

2D Strided to/from CSC, COO to CSC, CSC to CSC conversion. (#77521)

cpuhrsch committed 2 years ago

e10a002e

[reland][chalf] enable testing for multiple ops (#77656)

kshitij12345 committed 2 years ago

687ab973

add native view_copy.out ops, teach codegen about tensorlist out=

bdhirsh committed 2 years ago

edc904d6

[flaky test reporting] print stack trace for flaky reruns (#77664)

janeyx99 committed 2 years ago

a325fa94

[primTorch] Enforces stride metadata (#77542)

Mike Ruberry committed 2 years ago

580a0538

[FSDP][Easy] Remove extraneous print

awgu committed 2 years ago

93b20b02

Support str for Sparse Compressed tensors

pearu committed 2 years ago

ccc991ba

Migrate cross compilation trunk test to use macos12 to build MPS

albanD committed 2 years ago

6436fba3

Migrate x86 trunk build/test to macos12

albanD committed 2 years ago

7c3d3b75

Fix derivatives for linalg.vector_norm(..., dtype=)

lezcano committed 2 years ago

13d8fb93

Newer Older