strisunshinewentingwang/pytorch

Pull Requests Commits

[DCP][state_dict] Implement pin_memory and shared_memory copy for _offload_state_dict_to_cpu (#120378)

fegin committed 199 days ago

5abf7972

Remove unnecessary const_casts (#121225)

cyyever committed 199 days ago

6ecd6588

[export] Ensure optional fields always have default value. (#121163)

zhxchen17 committed 199 days ago

85c807b3

[dynamo] Fix handling of invalid args (#121110)

jansel committed 199 days ago

35004b8a

[dynamo] Remove extra guard for tensor constant attrs (#121106)

jansel committed 199 days ago

4f19b5f7

Disable remote cache test on ROCM (#121210)

oulgen committed 199 days ago

e4352182

Fix lintrunner-noclang (#121205)

angelayi committed 199 days ago

f25a25fd

Update Triton (#119457)

bhack committed 199 days ago

fbf36d01

Spectral norm value test (#121068)

redwrasse committed 199 days ago

59d9f1e2

Add exhaustive module and optimizer tests for torch.load(state_dict, weights_only=True) (#121049)

mikaylagawarecki committed 199 days ago

d621e3e3

[ATen][Native][CUDA] Decrease max_threads in ctc_loss (#120746)

Aidyn-A committed 199 days ago

42821d46

Fix make triton command on release branch (#121169)

atalman committed 199 days ago

12191f4b

skip detectron2_fcos_r_50_fpn in dynamic shape test (#120697)

jiayisunx committed 199 days ago

ee557d8f

Temporarily increased compile time limit of #GPUs to 120. (#121076)

tringwald committed 199 days ago

c4a15708

[FSDP][StateDict] Allow FULL_STATE_DICT option for 2D (#120837)

wz337 committed 199 days ago

de8af280

[CUDACachingAllocator] Turn Allocator::allocate into non-const (#120969)

cyyever committed 199 days ago

507611f9

[Dynamo] Fix inspect.getattr_static doesn't work well for torch.utils._cxx_pytree.PyTreeSpec (#120812)

yanboliang committed 199 days ago

46c9d646

Fix README Typo (#120892)

leimao committed 199 days ago

311cc564

[hoo] Add with_effects to handle side effectful ops (#120296)

angelayi committed 199 days ago

a7e93c34

Make configs hash part of remote cache key (#121152)

oulgen committed 199 days ago

29976519

Correctly read the cache key for remote cache (#121151)

oulgen committed 199 days ago

43416e30

Move JK check to on-demand (#121182)

oulgen committed 199 days ago

9e166223

Remove ids_of_folded_args from test_triton_kernel_equal_to_1_arg (#121192)

aakhundov committed 199 days ago

9ccff0af

[export][reland] Disable exported_program.__call__ (#120019)

angelayi committed 199 days ago

4b49bc19

[AOTI] Update cpp wrapper codegen to use v2 C shim (#120714)

desertfire committed 199 days ago

6ddf5cf8

[AOTI] Use torchgen to generate C shim functions (#120513)

desertfire committed 199 days ago

bd19d6d8

[ATen-vulkan] Implement global shader registry (#121088)

SS-JIA committed 199 days ago

ffe45a81

Update torchbench pin (#121029)

angelayi committed 199 days ago

c3c618c7

Fix dynamo failure (#121167)

tugsbayasgalan committed 199 days ago

a15c0256

Revert "Update Triton (#119457)"

pytorchmergebot committed 199 days ago

3381f282

Newer Older