OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-08 07:39:33 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
64,995 Commits 8,304 Branches 150 Tags 1.5 GiB
3058700f7f
Commit Graph

60 Commits

Author SHA1 Message Date
Jason Ansel
e071d72f3c Tag dynamo backends as debug/experimental (#93878) 
Hides debug/experimental backends by default.

Before:
```
torch._dynamo.list_backends()
['aot_eager', 'aot_eager_decomp_partition', 'aot_torchxla_trace_once', 'aot_torchxla_trivial', 'aot_ts', 'aot_ts_nvfuser', 'cudagraphs', 'dynamo_accuracy_minifier_backend', 'dynamo_minifier_backend', 'eager', 'inductor', 'ipex', 'nvprims_aten', 'nvprims_nvfuser', 'onnxrt', 'tensorrt', 'torchxla_trace_once', 'torchxla_trivial', 'ts', 'tvm']
```

After:
```
torch._dynamo.list_backends()
['aot_ts_nvfuser', 'cudagraphs', 'inductor', 'ipex', 'nvprims_nvfuser', 'onnxrt', 'tensorrt', 'tvm']
```

Fixes https://github.com/pytorch/pytorch/issues/93733

Pull Request resolved: https://github.com/pytorch/pytorch/pull/93878
Approved by: https://github.com/voznesenskym
 2023-02-04 00:50:51 +00:00
Jason Ansel
dfac113cfc Remove torch/_dynamo/optimizations (#93871) 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/93871
Approved by: https://github.com/voznesenskym
 2023-02-03 21:54:28 +00:00
Jason Ansel
5f4fec7459 Fix/refactor dynamo tvm backend (#93870) 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/93870
Approved by: https://github.com/shingjan, https://github.com/desertfire
 2023-02-03 21:48:31 +00:00
Jason Ansel
0a93e6db5a Fix/refactor dynamo ipex backend (#93863) 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/93863
Approved by: https://github.com/desertfire
 2023-02-03 21:42:27 +00:00
Jason Ansel
203b2cad3e Remove fx2trt/torch2trt backends (#93822) 
These backends have been broken for some time.  I tried to get them
running again, but as far as I can tell they are not maintained.
Installing torch_tensorrt downgrades PyTorch to 1.12.  If I manually
bypass that downgrade, I get import errors from inside fx2trt.  Fixes that
re-add these are welcome, but it might make sense to move these wrappers
to the torch_tensorrt repo once PyTorch 2.0 support is added.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/93822
Approved by: https://github.com/frank-wei
 2023-02-03 21:04:21 +00:00
Jason Ansel
5d709af59a Rename aot_cudagraphs to cudagraphs (#93821) 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/93821
Approved by: https://github.com/ezyang
 2023-02-03 21:01:27 +00:00
Jason Ansel
a5ff40032d Fix/refactor dynamo onnxrt backend (#93818) 
Fixes https://github.com/pytorch/pytorch/issues/90352

Pull Request resolved: https://github.com/pytorch/pytorch/pull/93818
Approved by: https://github.com/voznesenskym
 2023-02-03 20:48:02 +00:00
Jason Ansel
60e8c766b5 Refactor dynamo training backends (#93409) 
This splits training.py into many files and moves them from `dynamo.optimizations.training` to `dynamo.backends.*`.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/93409
Approved by: https://github.com/ezyang
 2023-02-03 03:07:15 +00:00
Jason Ansel
2b0d7e63f0 Move dynamo.optimizations.distributed to backends (#93408) 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/93408
Approved by: https://github.com/wconstab
 2023-02-02 20:42:17 +00:00
Jason Ansel
ee2729890c Refactor dynamo register_backend/BACKENDS (#93389) 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/93389
Approved by: https://github.com/voznesenskym
 2023-02-02 19:41:48 +00:00