James Reed
1d26a3ae7e
Open registration for c10 thread pool ( #17788 )
...
Summary:
1. Move ATen threadpool & open registration mechanism to C10
2. Move the `global_work_queue` to use this open registration mechanism, to allow users to substitute in their own
Pull Request resolved: https://github.com/pytorch/pytorch/pull/17788
Reviewed By: zdevito
Differential Revision: D14379707
Pulled By: jamesr66a
fbshipit-source-id: 949662d0024875abf09907d97db927f160c54d45
2019-03-08 15:38:41 -08:00
Ilia Cherniavskii
348d1889ff
Fix operator initialization order ( #15445 )
...
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15445
Initilize task graph after operators (task graph uses ops)
Reviewed By: yinghai
Differential Revision: D13530864
fbshipit-source-id: fdc91e9158c1b50fcc96fd1983fd000fdf20c7da
2019-02-26 15:41:16 -08:00
Ilia Cherniavskii
e9cd781681
Back out "Revert D13043261: [caffe2] Task graph and task future abstractions in executor"
...
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/15030
Reviewed By: bddppq
Differential Revision: D13408998
fbshipit-source-id: 9eb675e09fbc4829eab34df7aa660a0590816feb
2018-12-10 19:30:58 -08:00
Junjie Bai
4a145cd95c
Revert D13043261: [caffe2] Task graph and task future abstractions in executor
...
Differential Revision:
D13043261
Original commit changeset: d89424354aea
fbshipit-source-id: b307e3281c4d83b60ba2bfadcbcf69afb7a41412
2018-12-10 16:03:59 -08:00
Ilia Cherniavskii
029600813e
Task graph and task future abstractions in executor
...
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/14116
Reviewed By: dmudiger
Differential Revision: D13043261
fbshipit-source-id: d89424354aea14d1d14eb8320fb3aa34908a4e81
2018-12-10 14:28:56 -08:00