strisunshinewentingwang/pytorch

Pull Requests Commits

[inductor] generate triton kernel benchmark

shunting314 committed 1 year ago

a101f4ec

[ao][fx] Enable observed -> quantized float for static quantized MultiheadAttention (#95636)

kev-zheng committed 1 year ago

f1dbfe2f

Clean up unused `fill_` sample inputs (#95117)

qqaatw committed 1 year ago

fafb4109

Add missing f-string specifiers (#95707)

ezyang committed 1 year ago

835122c8

Add standalone torch._inductor.compile() API (#95594)

jansel committed 1 year ago

e13b8041

[quant][pt2e] Add support for dynamic quantization with symmetric quant for input (#94854)

JacobSzwejbka committed 1 year ago

fc324d34

[dynamo] avoid truncation of python pointers (#95619)

cyyever committed 1 year ago

f8ad64d5

[dtensor][BE] remove redundant tests (#94838)

wanchaol committed 1 year ago

1e15a272

[dtensor] support creating DTensor in submesh (#95458)

wanchaol committed 1 year ago

2a1cb964

[dtensor] refactor get_coordiniate (#95457)

wanchaol committed 1 year ago

261eb46d

[dtensor] use tracing for metadata prop (#95456)

wanchaol committed 1 year ago

bb9a05b1

Enabling FlashAttention for SDPA when given NestedTensor (#95438)

drisspg committed 1 year ago

80614783

Update skip message to reflect why test is being skipped (#95127)

mikekgfb committed 1 year ago

57f2c588

MHA torch.jit.script fix for in_proj_weight = None (#95653)

mikekgfb committed 1 year ago

4fada6eb

Add dynamo graph break stats to CI (#95635)

wconstab committed 1 year ago

1a727126

[MPS] Add pow.Scalar (#95201)

qqaatw committed 1 year ago

f33180fb

Add prelu into Autocast CPU whitelist (#95366)

yanbing-j committed 1 year ago

71ad1005

Reland #94719 - Update ideep to add primitive cache for ARM (#95688)

yanbing-j committed 1 year ago

b87229f1

[MTA] Skip size-0 tensors in `multi_tensor_apply` (#94655)

awgu committed 1 year ago

05943712

[MPS] Add copysign op. (#95552)

qqaatw committed 1 year ago

9e16f128

[MPS] Fix type casting copy with storage offset (#95573)

qqaatw committed 1 year ago

b7c2a651

[pt] add share_memory_ to aten TensorBase (#95557)

xunnanxu committed 1 year ago

7c66333c

Handle int/float arguments for cpp codegen in inductor (#95533)

ezyang committed 1 year ago

58648822

[bazel] enable sccache+nvcc in CI (#95528)

vors committed 1 year ago

447f5b5e

Update Dispatcher.cpp (#95589)

co63oc committed 1 year ago

49ba1196

Fix grammatical errors in contribution guide (#95454)

ajithvcoder committed 1 year ago

3944e7c3

Fix typos under torch/_dynamo directory (#95599)

kiszk committed 1 year ago

46385b3e

fix DeprecationWarning (#95545)

stas00 committed 1 year ago

38c32e19

Update dynamic skips (#95587)

ezyang committed 1 year ago

3762e801

[Inductor] Support sparse_grad for torch.gather (#95490)

desertfire committed 1 year ago

8b054338

Older