strisunshinewentingwang/pytorch

Pull Requests Commits

[JIT] Initialize CUDA context before launching fused kernel (#65064)

desertfire committed 2 years ago

719d0c6d

Wextra fix for Tensorshape.cpp (#66320)

r-barnes committed 2 years ago

8a02d3e5

Remove cuda/Loops.cuh dependency on native_functions.yaml (#64168)

peterbell10 committed 2 years ago

731cf494

Revert D31445799: [nnc] Use given kernel function name while emitting code

navahgar committed 2 years ago

92ce1885

Revert D31445797: [nnc] Added a cache to use singleton instances of PytorchLLVMJIT for every triple,cpu,attrs combination

navahgar committed 2 years ago

2e6fa026

Revert D31445798: [Static Runtime] Cleanup LLVMCodeGen memory after code gen completes

navahgar committed 2 years ago

097fdcdf

Remove Tensor.h include from TensorIterator.h (#64167)

peterbell10 committed 2 years ago

0be36d79

Migrate THCStorage_resizeBytes to ATen (#65944)

peterbell10 committed 2 years ago

bc1dec9b

Concatting multiple linear layers with same input Tensor (different weight/bias) (#63198)

Gamrix committed 2 years ago

3bad5406

[jit] Implement one-argument AliasDb::mayContainAlias more efficiently (#65177)

swolchok committed 2 years ago

94845fc4

[jit] Add cache for MemoryDAG::collectAllContainedMemoryLocations (#65122)

swolchok committed 2 years ago

c80693f7

[static runtime] Pre-allocate hash tables (#65343)

swolchok committed 2 years ago

3ef69a45

[LTC] Code-gen dropout (#66115)

alanwaketan committed 2 years ago

Verified 2d0fb6af

slow_conv3d grad_weight: call gemm directly (#65759)

peterbell10 committed 2 years ago

0020a151

log API usage for fsdp API in PyTorch (#64964)

zhaojuanmao committed 2 years ago

dfb64b32

Revert D31389480: [pytorch][PR] Allow external CUDA streams to be set as current

lw committed 2 years ago

201174cb

[PG Wrapper][BE] Add collective information when monitored barrier error is (#66167)

rohan-varma committed 2 years ago

b72a1782

[PG Wrapper][BE] Make some methods private (#66166)

rohan-varma committed 2 years ago

b5b1d49a

Move intraop_launch_future from Parallel.h (#64166)

peterbell10 committed 2 years ago

0cad2c06

[jit] Reduce refcounting of Types (#65345)

swolchok committed 2 years ago

2d885ab7

[jit] Refcounting spot fixes (#65346)

swolchok committed 2 years ago

1ae468a4

[DataPipe] moving mux IterDataPipe test to the right location (#66277)

NivekT committed 2 years ago

8ebe1a92

[DataPipe] adding test for IterableWrapperIterDataPipe (#66276)

NivekT committed 2 years ago

ed178516

[DataPipe] adding SequenceWrapperMapDataPipe (#66275)

NivekT committed 2 years ago

e808e3d3

quantized embedding: make error message clearer (#66051)

vkuzo committed 2 years ago

a7cc07f1

make error message when trying to quantize non floats more specific (#66050)

vkuzo committed 2 years ago

c9aba3b1

quantized add: enable broadcasting (#66049)

vkuzo committed 2 years ago

81660c08

Rename int to long, add more C++ types. (#66108)

ezyang committed 2 years ago

ece02218

Allow registration of custom symbolics for prim namespace (#64460) (#66139)

ezyang committed 2 years ago

11bc4356

[ONNX] Enable scripting tests (#64780) (#66138)

ezyang committed 2 years ago

9b09a5f7

Newer Older