strisunshinewentingwang/pytorch

Pull Requests Commits

[Nvfuser] Add cast support between double and half types

rdspring1 committed 2 years ago

33be4c94

Change mixed precision API

rohan-varma committed 2 years ago

10bf20da

[complex32] support printing the tensor

kshitij12345 committed 2 years ago

e36d25fb

Port functorch decomps over and fix some tests

Chillee committed 2 years ago

fb246140

Provide an auto wrap policy for common transformer models

zhaojuanmao committed 2 years ago

786903ea

Move functorch decompositions to PyTorch

ezyang committed 2 years ago

a3f10ec2

[fix] mul chalf: cpu scalar and cuda tensor case

kshitij12345 committed 2 years ago

1f1d0b30

Nvfuser faster fallback

jjsjann123 committed 2 years ago

100e72f5

Make Opinfo a dataclass to get a more useful repr

francescocastelli committed 2 years ago

201ddafc

Add better error message for FX when using concrete_args

Chillee committed 2 years ago

2540f866

[CUDA Graphs] Fix OOM inside graph capture_begin

fps7806 committed 2 years ago

b0c5fba9

Revert "[Re-landing 68111] Add JIT graph fuser for oneDNN Graph API (Preview4.1)"

pytorchmergebot committed 2 years ago

3dcd67a1

Allow sharding for distributed tests

janeyx99 committed 2 years ago

0708630d

[primTorch] Prototype nvFuser integration and test_prims.py

Mike Ruberry committed 2 years ago

fe1968de

Enable xla test

JackCaoG committed 2 years ago

87a05fa8

[Re-landing 68111] Add JIT graph fuser for oneDNN Graph API (Preview4.1)

chunyuan-w committed 2 years ago

8b11d810

[quant][core][gpu][improvement] Added support for padding quantized cudnn conv2d operator

dzdang committed 2 years ago

8c47e9dc

Add a matching lerp implementation to eager mode. (#1612)

jjsjann123 committed 2 years ago

ac31e5d4

Unsafe Sparse Compressed tensor factory function

pearu committed 2 years ago

ff10e459

Export aten::to("cpu") and aten::to(device="cpu")

Thiago Crepaldi committed 2 years ago

7afe4afd

Fix string repr for nested tensors & run nested tensor tests

jbschlosser committed 2 years ago

fd3cfb68

[GH1] Fix pending/not run conditional and add a way to test new merge_rules

janeyx99 committed 2 years ago

4c014a73

Elias Ellison committed 2 years ago

e5a55af3

Retry - [NVFuser] always use fallback if fusion fails

davidberard98 committed 2 years ago

e52dc988

Add torch.linalg.ldl_factor_ex and torch.linalg.ldl_solve

IvanYashchuk committed 2 years ago

8bb72030

[ONNX] Documentation and typing annotations in registry

justinchuby committed 2 years ago

2e841e68

Readme update to remove old python version

albanD committed 2 years ago

8880e3a3

Add test_jit_cuda_fuser.py to NVFuser merge rules

davidberard98 committed 2 years ago

5b6efb82

Nvfuser - Type Promotion Fix

rdspring1 committed 2 years ago

e9f17da2

[NVFuser] Reduce accuracy threshold for fp16 nvfuser opinfos

davidberard98 committed 2 years ago

4d527cd2

Newer Older