strisunshinewentingwang/pytorch

Pull Requests Commits

[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

Shard rocm distributed tests

janeyx99 committed 2 years ago

ad4f51d7

Disable XLA test workflow

janeyx99 committed 2 years ago

833d65ae

Add __all__ for torch.cuda.memory

anjali411 committed 2 years ago

9bf2c87e

Hipify fixes for a successful DeepSpeed build

rraminen committed 2 years ago

7422ccea

[quant][core][gpu][feature] Implemented quantized cuda adaptive average pool2d op (#76081)

dzdang committed 2 years ago

bbc263eb

Adding `broadcast_in_dim` and non-contiguous Tensor support NVFuser Python Frontend

kevinstephano committed 2 years ago

f51516df

[PT-D][Sharding] Move Partial Tensor to the _shard folder and add logic to remove padding (#76199)

fduwjj committed 2 years ago

c1037d0d

[quant] Fix tensorrt config after the backend_config_dict refactor (#76414)

jerryzh168 committed 2 years ago

b69d44da

[shard] Extensible Sharder and ShardingPlanner (#75844)

wanchaol committed 2 years ago

a37141ad

Revert "[JIT] Register decomp reland"

pytorchmergebot committed 2 years ago

a5bc02ae

Revert "[JIT] Add another decomposition api"

pytorchmergebot committed 2 years ago

2af0c42a

[quant][core][gpu][improvement] Converted reinterpret_cast<T *>(some_int8_tensor.data_ptr()) calls to some_int8_tensor.data_ptr<int8_t> in quantized cudnn operator files (#75980)

dzdang committed 2 years ago

9e73e890

Make lazy tensor ptr class customizable (#76476)

wconstab committed 2 years ago

d0cb31d5

Newer Older