strisunshinewentingwang/pytorch

Pull Requests Commits

create function to get ProcessGroupNCCL uid (#121132)

shengbao-zheng committed 196 days ago

60aaba41

[BE] Remove unnecessary requires_cuda in common_optimizers.py (#121249)

janeyx99 committed 196 days ago

83d095c2

Add capturable single tensor Adamax (#121183)

janeyx99 committed 196 days ago

53bdae73

Forward fix lint after 121202 (#121425)

clee2000 committed 197 days ago

af88425c

[export] update docs to not export raw functions (#121272)

suo committed 197 days ago

c3c15eb9

Revert "[ATen][CUDA][CUBLAS] cublasLtMatmul increase workspace_size (#120925)"

pytorchmergebot committed 197 days ago

862b99b5

[profiler] record nccl version in distributed info (#121044)

shengbao-zheng committed 197 days ago

eea37c6d

[Clang-tidy header][23/N] Enable clang-tidy coverage on aten/src/ATen/*.{cpp,h} (#121380)

cyyever committed 197 days ago

3aa512cd

[dynamo] relax missing symbols runtime assert (#121339)

IvanKobzarev committed 197 days ago

9a450019

[Inductor] Allocate another shard for testing cpp-wrapper JIT (#121310)

desertfire committed 197 days ago

0339f1ca

[Inductor] Enable ABI-compatible mode for cpp-wrapper JIT (#121309)

desertfire committed 197 days ago

7e598c00

[ Inductor ] Shape padding honors output stride preservation (#120797)

kadeng committed 197 days ago

57fc35a3

Change ATEN generator argument type to const std::optional<Generator>& (#120076)

cyyever committed 197 days ago

4305c64f

[inuctor] fix the layout problem for nll_loss2d_backward (#121173)

shunting314 committed 197 days ago

1ce50496

add int8 packed gemm support on CPU device (#118056)

mingfeima committed 197 days ago

b3065f68

[FSDP2] Relaxed check for parent mesh (#121360)

awgu committed 197 days ago

e8e3049f

Add SDPA pattern for HuggingFace models BF16 (#121202)

Valentine233 committed 197 days ago

db36d21f

Wrap remote cache creation with a try-catch (#121340)

oulgen committed 197 days ago

953c6c37

Modify StorageImplCreateHelper (#118459)

CLiqing committed 197 days ago

291ce86a

[Quant][Inductor] Fix q/dq per channel lowering with 64-bit qparams (#120984)

Xia-Weiwen committed 197 days ago

f848e9c6

[DTensor][XLA] refactor DTensor _xla API (#113214)

yeounoh committed 197 days ago

4f9d4e1a

[CUDACachingAllocator] Simplify update_stat and avoid casts (#120964)

cyyever committed 197 days ago

c723514e

Make CausalBias a torch.Tensor subclass again (#121358)

drisspg committed 197 days ago

55232c4e

[dtensor][debug] have visualize_sharding correctly print for sub-mesh DTensor (#121216)

XilunWu committed 197 days ago

df2ad1fe

[dtensor][1/N] add torchrec even row-wise sharding example (#120260)

XilunWu committed 197 days ago

77873f6f

[dtensor][debug] allow visualize_sharding to print header (#121179)

XilunWu committed 197 days ago

9cc0f23e

Bugfix consume_prefix_in_state_dict_if_present function to keep the order of the state_dict (#117464)

jmarintur committed 197 days ago

a2854ae9

Prevent infinite recursion within Tensor.__repr__ (#120206)

aorenste committed 197 days ago

edd80f87

graph break on sparse tensors constructions (#120458)

laithsakka committed 197 days ago

eb4d87f2

[TP] Introduce Sequence Parallel Style for Laynorm/RMSNorm/Dropout (#121295)

wanchaol committed 197 days ago

1a28ebff

Newer Older