strisunshinewentingwang/pytorch

Pull Requests Commits

Allow in graph einops operators (#99631)

wconstab committed 1 year ago

6427b849

Make testing._internal.common_utils safe to import (#99659)

wconstab committed 1 year ago

716ba685

[dynamo] Fix example_inputs with unsqueeze_ (#98696)

jansel committed 1 year ago

d168161c

[DTensor] Change Sharding algorithm to be in line with ``torch.chunk()`` (#98722)

wz337 committed 1 year ago

0d2b55c4

add storage serialization methods for privateuse1 (#98920)

XDaoHong committed 1 year ago

27f8eb8c

[inductor] Update triton pin (#99209)

peterbell10 committed 1 year ago

907f2dad

Disable SDPA FlashAttention backward and mem eff attention on sm86+ for head_dim above 64 (#99105)

cpuhrsch committed 1 year ago

fdeee436

Require at least one tensor to be marked dynamic with --dynamic-batch-only (#99620)

ezyang committed 1 year ago

fc8fa6c3

Reuse tracing context and fake tensors from backwards in forwards (#99619)

ezyang committed 1 year ago

abdd1f4a

bug-report.yml fix broken link (#99425)

Chizkiyahu committed 1 year ago

bbfd577b

Fix broken links in contribution_guide.rst (#99295)

simonseo committed 1 year ago

9f950321

[Dynamo] Merge symbolic_converter SETUP_WITH & BEFORE_WITH (#99651)

yanboliang committed 1 year ago

c9b08a08

Temporarily move ROCm to unstable (#99579)

kit1980 committed 1 year ago

c4120569

Fix buck parsing in OSS build (#99648)

malfet committed 1 year ago

37bcdb98

[quant][pt2] Add Conv + BN fusion for prepare QAT (#98568)

andrewor14 committed 1 year ago

22af604e

[reland][inductor] coordinate descent tuning upon max-autotune (#99594)

shunting314 committed 1 year ago

418a9fb9

Remove redundant `found_inf` recompute from `_step_supports_amp_unscaling` path (#98620)

crcrpar committed 1 year ago

b87c7ab6

Clarify error message of torch.nn.functional.embedding_bag (#99471)

tfsingh committed 1 year ago

a8e1893b

[dynamo] Support BUILD_MAP_UNPACK (#98664)

jansel committed 1 year ago

e68e84ef

[profiler] support cuLaunchKernel (for triton kernel launches) & update kineto submodule (#99571)

davidberard98 committed 1 year ago

c19d19f6

Skip some detectron2_maskrcnn models with KeyError _ignore_torch_cuda_oom (#99599)

huydhn committed 1 year ago

5315317b

Revert "Build Windows binaries with Visual Studio 2022 Build Tools (#90855) (#99591)

huydhn committed 1 year ago

7c3fa5c7

Stop testing if CUDA is initialized on teardown (#99627)

ezyang committed 1 year ago

0a98289a

Improve torch.cond useability: Return UserError with actionable error messages (#98909)

guangy10 committed 1 year ago

aa4ed332

Guard on default device (#99551)

ezyang committed 1 year ago

e47e8c9d

[SPMD] Allow users to dynamically pass the last_iter to IterGraphModule (#99575)

fegin committed 1 year ago

88c45a19

Fallback for Complex Dtypes in Inductor (#97198)

eellison committed 1 year ago

7acb0bdd

Turn on meta converter for complex (#98869)

eellison committed 1 year ago

638feec4

Allow getting type of ScriptObject (#99542)

lw committed 1 year ago

df84d740

Reland of "Python binding to set/get CUDA rng state offset" (#99565)

anijain2305 committed 1 year ago

971df458

Newer Older