OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
53,290 Commits 8,304 Branches 150 Tags 1.5 GiB
bdb14238ec
Commit Graph

23 Commits

Author SHA1 Message Date
AllenTiTaiWang
bdb14238ec [Reland][ONNX] Move all torch.onnx.export related tests to test/onnx (#87292) 
Moving torch.onnx.export related tests to test/onnx integrates ONNX tests to the same CI machine, so the testing environment can be better managed.

Fixes https://github.com/pytorch/pytorch/issues/87320
Pull Request resolved: https://github.com/pytorch/pytorch/pull/87292
Approved by: https://github.com/thiagocrepaldi, https://github.com/BowenBao, https://github.com/kit1980, https://github.com/malfet
 2022-11-01 14:22:46 +00:00
PyTorch MergeBot
e9599724fa Revert "[ONNX] Move all torch.onnx.export related tests to test/onnx (#87292)" 
This reverts commit e3e84830aa.

Reverted https://github.com/pytorch/pytorch/pull/87292 on behalf of https://github.com/weiwangmeta due to breaking internal test relating to quantization eager tests, see test/quantization/eager/test_quantize_eager_ptq.py test_lower_graph_linear and test_lower_graph_conv2d
 2022-10-31 19:55:58 +00:00
AllenTiTaiWang
e3e84830aa [ONNX] Move all torch.onnx.export related tests to test/onnx (#87292) 
Moving torch.onnx.export related tests to test/onnx integrates ONNX tests to the same CI machine, so the testing environment can be better managed.

Fixes https://github.com/pytorch/pytorch/issues/87320
Pull Request resolved: https://github.com/pytorch/pytorch/pull/87292
Approved by: https://github.com/thiagocrepaldi, https://github.com/BowenBao, https://github.com/kit1980
 2022-10-29 05:31:30 +00:00
Jane Xu
09c7771e9c Set test owners for jit tests (#66808) 
Summary:
Action following https://github.com/pytorch/pytorch/issues/66232

Pull Request resolved: https://github.com/pytorch/pytorch/pull/66808

Reviewed By: mrshenli

Differential Revision: D31761414

Pulled By: janeyx99

fbshipit-source-id: baf8c49ff9c4bcda7b0ea0f6aafd26380586e72d
 2021-10-25 07:51:10 -07:00
Jane Xu
b1bae01e0c Replace raiseRegex with raiseRegexWithHighlight test_async.py (#55489) 
Summary:
Step to resolving https://github.com/pytorch/pytorch/issues/55072

Pull Request resolved: https://github.com/pytorch/pytorch/pull/55489

Reviewed By: gmagogsfm

Differential Revision: D27625872

Pulled By: janeyx99

fbshipit-source-id: 1ee606a30b13d041d8d107e6cc23be16c076d072
 2021-04-07 14:30:23 -07:00
Richard Barnes
8c25b9701b Type annotations in test/jit (#50293) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/50293

Switching to type annotations for improved safety and import tracking.

Test Plan: Sandcastle tests

Reviewed By: xush6528

Differential Revision: D25853949

fbshipit-source-id: fb873587bb521a0a55021ee4d34d1b05ea8f000d
 2021-01-12 16:47:06 -08:00
Heitor Schueroff
1bb7d8ff93 Revert D25717504: Clean up some type annotations in test/jit 
Test Plan: revert-hammer

Differential Revision:
D25717504 (a4f30d48d8)

Original commit changeset: 9a83c44db02e

fbshipit-source-id: e6e3a83bed22701d8125f5a293dfcd5093c1a2cd
 2021-01-08 12:14:48 -08:00
Richard Barnes
a4f30d48d8 Clean up some type annotations in test/jit (#50158) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/50158

Upgrades type annotations from Python2 to Python3

Test Plan: Sandcastle tests

Reviewed By: xush6528

Differential Revision: D25717504

fbshipit-source-id: 9a83c44db02ec79f353862255732873f6d7f885e
 2021-01-08 10:56:55 -08:00
Richard Barnes
ec6d29d6fa Drop unused imports from test (#49973) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/49973

From
```
./python/libcst/libcst codemod remove_unused_imports.RemoveUnusedImportsWithGlean --no-format caffe2/
```

Test Plan: Standard sandcastle tests

Reviewed By: xush6528

Differential Revision: D25727350

fbshipit-source-id: 237ec4edd85788de920663719173ebec7ddbae1c
 2021-01-07 12:09:38 -08:00
Meghan Lele
79108fc16c [JIT] Improve Future subtype checking (#44570) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/44570

**Summary**
This commit improves subtype checking for futures so that
`Future[T]` is considered to be a subtype of `Future[U]` if `U` is a
subtype of `V`.

**Test Plan**
This commit adds a test case to `test_async.py` that tests this.

Test Plan: Imported from OSS

Reviewed By: eellison

Differential Revision: D23660588

Pulled By: SplitInfinity

fbshipit-source-id: b606137c91379debab91b9f41057f7b1605757c5
 2020-09-16 18:54:51 -07:00
Dmytro Dzhulgakov
47e489b135 Make ExtraFilesMap return bytes instead of str (#43241) 
Summary:
In case we want to store binary files using `ScriptModule.save(..., _extra_files=...)` functionality. With python3 we can just use bytes only and not bother about it.

I had to do a copy-pasta from pybind sources, maybe we should upstream it, but it'd mean adding a bunch of template arguments to `bind_map` which is a bind untidy.

Let me know if there's a better place to park this function (it seems to be the only invocation of `bind_map` so I put it in the same file)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/43241

Reviewed By: zdevito

Differential Revision: D23205244

Pulled By: dzhulgakov

fbshipit-source-id: 8f291eb4294945fe1c581c620d48ba2e81b3dd9c
 2020-08-28 19:11:33 -07:00
Elias Ellison
f000b44d89 Fork/Join Inline Docs (relanding) (#40438) 
Summary:
Added fork/wait to docs/source/jit.rst, hopefully that will fix test error.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/40438

Differential Revision: D22188152

Pulled By: eellison

fbshipit-source-id: c19277284455fb6e7c0138b0c1423d90b147d18e
 2020-06-23 13:25:51 -07:00
Yanli Zhao
6c40ec55df Revert D22165477: [pytorch][PR] [JIT] Fork/Join inline docs 
Test Plan: revert-hammer

Differential Revision:
D22165477

Original commit changeset: 93132cd6987f

fbshipit-source-id: f3d5d35b6640d786ec3bada1396b5d7ad645c26d
 2020-06-22 20:51:56 -07:00
Elias Ellison
0d0608532c [JIT] Fork/Join inline docs (#39952) 
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/39952

Differential Revision: D22165477

Pulled By: eellison

fbshipit-source-id: 93132cd6987fdd2484112a57ef17912b8fcc5fab
 2020-06-22 15:34:05 -07:00
James Reed
56289ba31f [JIT] Improve error message when type annotation Future without a contained type (#39751) 
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/39751

Test Plan: Imported from OSS

Reviewed By: eellison

Differential Revision: D21960368

Pulled By: jamesr66a

fbshipit-source-id: 8650d31ff8070b12672c8d4b0224d4e69f619938
 2020-06-09 16:55:13 -07:00
Nikita Shulga
8811e4d00d Add/fix typing annotations to some functions (#39075) 
Summary:
Add missing typing imports to some jit tests
Add typing annotations to `torch.testing._compare_scalars_internal` and `torch.testing._internal.assertTrue`
Pull Request resolved: https://github.com/pytorch/pytorch/pull/39075

Differential Revision: D21882468

Pulled By: malfet

fbshipit-source-id: dd9858eb8e11a38411544cc64daf36fced807d76
 2020-06-04 13:40:04 -07:00
Will Constable
6fd48e24f1 Add support, test for kwargs in jit._fork (#38357) (#38665) 
Summary:
Closing 38357
Pull Request resolved: https://github.com/pytorch/pytorch/pull/38665

Reviewed By: suo

Differential Revision: D21643697

Pulled By: wconstab

fbshipit-source-id: c292c037f87bc2bb69a4ca163d7107d5396c53a2
 2020-05-19 13:02:46 -07:00
Wanchao Liang
c38dcd45d7 [jit] fix return different types bug in tracing module calls (#37190) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/37190

if module call return different types, we need to record them correctly

Test Plan: Imported from OSS

Differential Revision: D21214871

Pulled By: wanchaol

fbshipit-source-id: 46ba98f08ed4ade22f9740cb3fca84b29557e125
 2020-04-24 14:49:28 -07:00
Wanchao Liang
5362a0b948 [jit] fix lifting bug in tracing module calls (#37189) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/37189

This fix bug in tracing module calls to correct lift values with its
correponding value type, rather than the default tensor type.

Test Plan: Imported from OSS

Differential Revision: D21214872

Pulled By: wanchaol

fbshipit-source-id: f635154851365e2d7b88186d6e47634123eac42f
 2020-04-24 14:47:54 -07:00
Shihao Xu
cae6bdf199 [JIT] Mark aten::wait as having side effect, since it can represent RPC message received (#35695) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/35695

aten::wait was optimized out, causing RPC futures are not waited on.

Test Plan:
```
buck test mode/dev-nosan //caffe2/torch/fb/distributed/model_parallel/tests:test_dist_optim
```

```
buck test mode/dev-nosan //caffe2/test/distributed/rpc/jit:rpc_fork

buck build mode/dev-nosan //caffe2/test/distributed/rpc/jit:rpc_fork && \
buck-out/gen/caffe2/test/distributed/rpc/jit/rpc_fork\#binary.par \
-r test_python_future_with_jit
```

```
buck build mode/dev-nosan //caffe2/test:jit && \
buck-out/gen/caffe2/test/jit\#binary.par -r test_trace_fork_wait_inline
```

```
buck build mode/dev-nosan //caffe2/test:jit && \
buck-out/gen/caffe2/test/jit\#binary.par -r test_trace_fork_wait_inline_onnx
```

Differential Revision: D9562716

fbshipit-source-id: 35b2c971efa42949ffdf0910bd75a927eee8d965
 2020-03-31 22:17:25 -07:00
Wanchao Liang
4a84ac5f5d [jit] make Future type annotation available in Python (#27637) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/27637

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

Test Plan: Imported from OSS

Differential Revision: D20626866

fbshipit-source-id: 20d6a3a719fddcb33e0e17a56d7123535fa20d65
 2020-03-24 14:36:05 -07:00
Pritam Damania
f050b16dd9 Move pytorch distributed tests to separate folder for contbuild. (#30445) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/30445

Create distributed and rpc directories under caffe/test for better management
of unit tests.

Differential Revision: D18702786

fbshipit-source-id: e9daeed0cfb846ef68806f6decfcb57c0e0e3606
 2020-01-22 21:16:59 -08:00
Michael Suo
cc457ca30f split remaining "easy" tests (#29249) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/29249

This splits out all the tests that are "easy", leaving `TestJit`,
`TestScript`, the autogenerated tests, and a small docs test.

Splitting those into reasonable chunks is more effort which is less
mechanical.

Differential Revision: D18339007

Test Plan: Imported from OSS

Pulled By: suo

fbshipit-source-id: 69164b9f9a2c379fe8923a846c98dd3c37ccb70e
 2019-11-06 13:23:01 -08:00